在Tensorflow中自定义Softmax激活功能

时间:2020-09-28 09:46:20

标签: python tensorflow neural-network customization softmax

我在张量流中建立了一个神经网络,该神经网络应该预测事件发生的概率。 因此,作为输出函数,我通过了softmax。 我尝试自定义softmax函数,以便所有(三个)类的概率总和为1.05,最小(最大)概率为0.02(0.99)。 到目前为止,我有以下代码,但是我不确定这是否正确

def custom_activation(x):
    return K.softmax(x)*(1 + 0.05)

有人知道如何实现吗?

0 个答案:

没有答案