【问题标题】:How to get confusion matrix in pytorch?如何在pytorch中获得混淆矩阵?
【发布时间】:2021-03-09 01:07:34
【问题描述】:

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

【问题讨论】:

    标签: python pytorch


    【解决方案1】:

    如果您有目标类 y_test 和预测类 y_pred,您可以尝试使用 sklearn 包。

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

    【讨论】:

      猜你喜欢
      • 2021-08-01
      • 2020-02-23
      • 2021-12-08
      • 2021-10-25
      • 2020-09-20
      • 2019-12-28
      • 2021-12-03
      • 2018-11-09
      • 2021-11-30
      相关资源
      最近更新 更多