Torch Confusion matrix pour critère binaire

Question

Comment utiliser la matrice de confusion des packages optim avec BCECriterion, qui produit une sortie binaire? La dernière couche de mon modèle est nn.logsigmoid. Merci.

Answer 1

La matrice de confusion ne peut pas accepter les 0, elle veut les indices de classe de la cible et la prédiction.