我正在尝试使用来自 sklearn 0.22 的 plot_confusion_matrix 函数
但我的数字显示为 1.3e+0.2 格式
from sklearn.metrics import plot_confusion_matrixplt.figure(figsize=(100, 230))plot_confusion_matrix(clf, X, y, cmap=plt.cm.Blues)plt.xlabel('真实标签')plt.ylabel('预测标签')
我该如何修复这个问题?
回答:
使用函数中的 values_format 参数。 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 相同。