我现在正在处理一个多标签分类任务。
使用BCEWithLogitsLoss作为损失函数。学习结束,通过每个epoch中的train_loss和validate_loss来看,模型没有过拟合,并且loss稳定在了一个定值周围,应该不存在under fitting。
但是,通过sklearn的多标签confusion matrix函数获取的混淆矩阵,得到了类似于如下的矩阵:
735,0
265,0
997,0
3 ,0
几乎所有类别的True Positive都是0
这就非常奇怪了
当然,模型的输出使用了sigmoid以后再使用0.5作为分类的threshold。