这是 Keras 模型的最后一层。
model.add(Dense(3, activation='softmax')) model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['accuracy'])
我知道softmax层的输出是一个数组,概率总和为1,比如[0.1, 0.4, 0.5]
.
我有一个关于使用准确性作为指标的问题。
例如,当真实类别为[0, 0, 1]
且预测概率为 时[0.1, 0.4, 0.5]
,即使0.5
是最大概率,此预测的准确度也应为0
,因为0.5 != 1
. 那是对的吗?
更一般地,当输出层激活为 时softmax
,我们通常会得到浮动概率预测,并且在极少数情况下我们会得到 [0, 0, 1]
像 所以我们在使用as activation的时候不能用accuracy
as metric softmax
。那是对的吗?
墨色风雨
相关分类