luchi007/RNN_Text_Classify

多分类问题怎么设置labels

Opened this issue · 1 comments

你好,请问多分类问题怎么设置labels(targets), 比如说 二分类设置为0,1,三分类怎么设置labels , -1,0,1吗

如果最后一层使用softmax的话,label可以设置成多标签的分布,比如三类标签[0.5,0.5,0]表示分为前两个类,如果最后一层使用sigmoid的话,label可以设置成多个0,1向量,比如[1,1,0]可以表示结果分为前两个类,都可以使用交叉熵作为损失函数