python - 如何在pytorch中获得混淆矩阵？

Question

有15个班。我有两个张量，一个有目标类，另一个有预测类。如何获得混淆矩阵？

标签： pythonpytorch

Answer 1

假设您有目标类 y_test 和预测类 y_pred，您可以尝试使用 sklearn 包。

from sklearn.metrics import confusion_matrix
print(confusion_matrix(y_test, y_pred))