keras LSTM中的合计输出

时间:2018-04-19 11:12:33

标签: python keras lstm

也许这是一个虚拟问题,但我想知道我是否可以使用我的输出向量来满足特定要求。

我有多个输出(不是二进制),我希望它们的总和为1.我希望我的模型进行某种评估,以便我的输出向量满足这个要求。

1 个答案:

答案 0 :(得分:0)

你可以:

  • 在模型的末尾添加softmax激活,或
  • 创建自定义规范化功能

<强>使用SoftMax:

只需在模型末尾添加Activation('softmax') 这将执行一些日志操作,并可能添加一些您不想要的额外条件。

自定义规范化:

只需添加此图层:

import keras.backend as K

Lambda(lambda x: x / K.sum(x), output_shape=optional_with_tensorflow)