二元准则的火炬混淆矩阵

时间:2016-02-03 11:18:57

标签: machine-learning torch confusion-matrix criterion

你如何使用优化套餐'混淆矩阵与BCECriterion,产生二进制输出?我模型的最后一层是nn.logsigmoid。感谢。

1 个答案:

答案 0 :(得分:2)

混淆矩阵不能接受0,它想要目标的类索引和预测。