可视化分类器预测，分散

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

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

描述here的Sklearn混淆矩阵确实满足您的需求。