调整 ConfusionMatrixDisplay 的大小 (Scikit Learn)
Adjust size of 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()
上面的代码显示了这个图,太紧了:
你可以先发一个matplotlib.axes
object to the .plot
method of sklearn.metrics.ConfusionMatrixDisplay
. Set the size of the figure in matplotlib.pyplot.subplots
。
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)
如何设置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()
上面的代码显示了这个图,太紧了:
你可以先发一个matplotlib.axes
object to the .plot
method of sklearn.metrics.ConfusionMatrixDisplay
. Set the size of the figure in matplotlib.pyplot.subplots
。
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)