machine-learning - 二元标准的火炬混淆矩阵

Question

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

Answer 1

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