调整ConfusionMatrixDisplay的大小(ScikitLearn)
如何设置 ScikitLearn 的 ConfusionMatrixDisplay 绘制的图形的大小？
import numpy as np
from sklearn.metrics import ConfusionMatrixDisplay, confusion_matrix
cm = confusion_matrix(np.arange(25), np.arange(25))
cmp = ConfusionMatrixDisplay(cm, display_labels=np.arange(25))
cmp.plot()
上面的代码显示了这个图，太紧了：
回答
您可以将ax参数发送到plot现有 PyPlot 图的 。之前只设置图形的大小。
import numpy as np
from sklearn.metrics import ConfusionMatrixDisplay, confusion_matrix
import matplotlib.pyplot as plt
cm = confusion_matrix(np.arange(25), np.arange(25))
cmp = ConfusionMatrixDisplay(cm, display_labels=np.arange(25))
fig, ax = plt.subplots(figsize=(10,10))
cmp.plot(ax=ax)
THE END
二维码