我正在训练神经网络来预测给定输入的多个标签。我的输入是一个 200 大小的整数向量,输出应该是一个大小为 28 的布尔向量。我y
在示例对应的相应类上有一个 1,即y
应该看起来像:
[0, 0, 1, 0, ... 1, 0, 0]
。
现在,我在最后一层使用了 sigmoid 函数,并且我正在使用二元交叉熵进行训练,因为我希望我的模型将每个类视为彼此独立。
当我拟合我的模型时,我得到了相当不错的分类准确度,即 ~75%,但我想知道 Keras 用来表示示例中存在或不存在类的阈值,我的意思是,这是它决定的阈值:
prediction[prediction >= threshold] = 1
prediction[prediction < threshold] = 0