sklearn.metrics.classification_report 支持字段显示错误的标签编号

Question

我正在使用 sklearn.metrics.classification_report 来评估我的分类结果。

y_pred = np.argmax(model.predict(X_test), axis=1)
y_true = np.argmax(y_test, axis=1)
print(classification_report(y_true, y_pred, target_names=list(le.classes_)))

这是我的结果：

              precision    recall  f1-score   support

 Technology       0.00      0.00      0.00         1
 Travel           0.00      0.00      0.00         5
 Fashion          0.00      0.00      0.00        25
 Entertainment    0.72      1.00      0.84       130
 Art              0.00      0.00      0.00         7
 Politic          0.00      0.00      0.00        12

 avg / total       0.52      0.72      0.61       180

问题实际上是我有 7 个标签。顺序是科技、旅游、时尚、娱乐、艺术、政治、体育。实际上，我的 y_true 结果中没有任何艺术标签，但报告按顺序列出，它列出了艺术，但跳过了运动。它写政治为艺术的结果，体育的结果进入政治的行。

为什么它不跳过艺术？我不知道我该如何解决这个问题。

Answer 1

分类报告中的索引标签是参数“target_names”的值。请确保您为该论点提供了正确的值。

根据您需要的输出，您应该有 target_names = [“科技”、“旅游”、“时尚”、“娱乐”、“政治”、“体育”]

我建议，请检查“le.classes_”的输出，我不确定“le”指的是哪个转换器。