python - TypeError: plot_confusion_matrix() 得到了一个意外的关键字参数“标题”

Question

`import matplotlib.pyplot as plt  
 from sklearn.datasets import make_classification
 from sklearn.metrics import plot_confusion_matrix
 from sklearn.model_selection import train_test_split
 from sklearn.svm import SVC
 cm_plot_labels = ['Normal', 'Tuberculosis']
 cm_plot_title =['Confusion Matrix']
 plot_confusion_matrix(cm, cm_plot_labels, title='Confusion Matrix')

TypeError: plot_confusion_matrix() got an unexpected keyword argument 'title' 请告诉我如何解决该错误

标签： pythonmatplotlibscikit-learnsklearn-pandasmatplotlib-animation

Answer 1

您可以在使用时添加标题plot_confusion_matrix，如下所示：

clf = SVC(random_state=0)
clf.fit(X_train, y_train)
cm = plot_confusion_matrix(clf, X_test, y_test)
cm.ax_.set_title('Confusion Matrix')