Confusion Matrix Scikit-learn in Chapter 4

[ISTTeo]

Chapter 4 - Confusion Matrix

ImportError                               Traceback (most recent call last)
Cell In[21], line 1
----> 1 from sklearn.metrics import plot_confusion_matrix

ImportError: cannot import name 'plot_confusion_matrix' from 'sklearn.metrics'

The library was updated.

[ISTTeo]

We just need to import the new ConfusionMatrixDisplay:

from sklearn.metrics import ConfusionMatrixDisplay

cm = confusion_matrix(test_labels, result)
disp = ConfusionMatrixDisplay(confusion_matrix=cm, display_labels=MNB.classes_)
disp.plot()