我的标签看起来像这样
label = [0, 1, 0, 0, 1, 0]
意味着匹配的样本输入中存在类1, 4
。
答案 0 :(得分:1)
没有充分的理由为此创建one-hot
编码版本,如果您希望保持输出标签大小完全相同,在您的情况下为6
,你不能做one-hot
编码版本。
要完成multi-label classification
,您不能(更恰当地不应)使用softmax
作为激活。 Softmax
适用于只有一个输出可以是真值的情况。因此,在您的情况下,最好使用sigmoid cross-entropy
。