可视化分类器预测，分散

Question

我想绘制这样的散点图，您可以在其中看到分类器的错误预测和真实预测，包括预测的数量

到目前为止，我一直在尝试使用此散点图

predictions_rf = rf.predict(X_test)
plt.figure(figsize=(10,10))
plt.gca().set_aspect('equal', adjustable='box')
plt.scatter(Y_test,predictions_rf)
plt.xlabel("True Values")
plt.ylabel("Predictions")

我得到了这样的散点图： 也有可能看到错误的估计，但不是对角线，也不是模型预测正确和错误的数字

我如何改进我的代码以获得与第一个类似的解决方案？

谢谢

Answer 1

您要绘制的是一个混淆矩阵。

描述的 Sklearn 混淆矩阵 here 完全符合您的要求。