【发布时间】:2021-03-09 01:07:34
【问题描述】:
有 15 个班级。我有两个张量,一个有目标类,另一个有预测类。 如何获得混淆矩阵?
【问题讨论】:
有 15 个班级。我有两个张量,一个有目标类,另一个有预测类。 如何获得混淆矩阵?
【问题讨论】:
如果您有目标类 y_test 和预测类 y_pred,您可以尝试使用 sklearn 包。
from sklearn.metrics import confusion_matrix
print(confusion_matrix(y_test, y_pred))
【讨论】: