二元标准的 Torch 混淆矩阵

Question

如何将 optim 包的混淆矩阵与生成二进制输出的 BCECriterion 一起使用？我模型的最后一层是 nn.logsigmoid。谢谢。

Answer 1

混淆矩阵不能接受 0，它需要目标和预测的 class 索引。