声称报告

我们还有一个scikit-learn的功能, 可以直接打印我们的精确率, 召回率和F1分数. 让我们看看如何做到这一点.

怎么做...?

• 添加如下代码到python文件:

y_true = [1, 0, 0, 2, 1, 0, 3, 3, 3]
y_pred = [1, 1, 0, 2, 1, 0, 1, 3, 3]
confusion_mat = confusion_matrix(y_true, y_pred)
plot_confusion_matrix(confusion_mat)

# Print classification report
target_names = ['Class-0', 'Class-1', 'Class-2', 'Class-3']
print (classification_report(y_true, y_pred, target_names=target_names))

• 运行结果如下:

不用单独计算这些指标, 您可以直接使用此函数从模型中提取这些统计数据.