Keras - 范围内数字的目标函数[0-100]

时间:2016-07-24 21:41:41

标签: python neural-network keras

让我说我在[0-100]范围内的数字的基础事实(输出)。 是否有可能学会根据输入预测一个最小化原始数字(gt)的增量的数字?

Keras的目标类型在http://keras.io/objectives/

1 个答案:

答案 0 :(得分:1)

我认为你最好的选择是使用均方误差(loss='mse'),它会根据它们与地面实况的差异的平方来惩罚预测。您还需要为最后一层使用线性激活(默认)。

如果您特别关注将预测保持在[0,100]范围内,您可以创建一个修改后的目标函数,该函数会对[0,100]之外的预测进行甚至超过二次方的惩罚,但是可能没有必要,您可以使用np.clip(predictions, 0, 100)来剪切预测。