Confusion Matrix-Sklearn 0.22 数字格式错误

Question

我正在尝试使用来自 sklearn 0.22

的 plot_confusion_matrix

但是我的数字格式是 1.3e+0.2

from sklearn.metrics import plot_confusion_matrix
plt.figure(figsize=(100, 230))
plot_confusion_matrix(clf, X, y, cmap=plt.cm.Blues)
plt.xlabel('True label')
plt.ylabel('Predicted label')

我该如何解决这个问题？

Answer 1

在函数中使用值格式参数。 https://scikit-learn.org/stable/modules/generated/sklearn.metrics.plot_confusion_matrix.html

这里有一个使用values_format论证的例子：

disp = plot_confusion_matrix(classifier, X_test, y_test,
                                 display_labels=class_names,
                                 cmap=plt.cm.Blues,
                                 normalize=normalize,
                                   values_format="d")

如果你想只显示一个浮点数，你可以使用下面的格式'.1f'。它与 seaborn 中使用的 values_format 相同。