plot_confusion_matrix() got an unexpected keyword argument 'display_labels'
时间: 2023-10-15 13:02:39 浏览: 282
plot_ps.zip_IDL plot_IDL plot 界面
这个错误通常是因为你正在使用一个较旧版本的scikit-learn,而`display_labels`参数是在较新的版本中引入的。你可以尝试更新scikit-learn到最新版本,或者在使用`plot_confusion_matrix()`时省略`display_labels`参数。如果你需要使用`display_labels`参数,可以通过以下方式解决该问题:
1. 更新scikit-learn
使用以下命令更新scikit-learn:
```
pip install --upgrade scikit-learn
```
2. 省略`display_labels`参数
在使用`plot_confusion_matrix()`时省略`display_labels`参数,例如:
```python
plot_confusion_matrix(model, X_test, y_test)
```
3. 手动指定参数
如果你正在使用较旧的scikit-learn版本,你可以手动指定`display_labels`参数,例如:
```python
from sklearn.metrics import ConfusionMatrixDisplay
ConfusionMatrixDisplay(confusion_matrix=matrix, display_labels=labels).plot()
```
这个方法使用`ConfusionMatrixDisplay`类而不是`plot_confusion_matrix`函数来绘制混淆矩阵。
阅读全文