标签: machine-learning torch confusion-matrix criterion
你如何使用优化套餐'混淆矩阵与BCECriterion,产生二进制输出?我模型的最后一层是nn.logsigmoid。感谢。
答案 0 :(得分:2)
混淆矩阵不能接受0,它想要目标的类索引和预测。