In [1]:

    

from sklearn.metrics import confusion_matrix
import seaborn as sns
import matplotlib.pyplot as plt


    

In [2]:

    

y_true = [0, 0, 0, 0, 0, 1, 1, 1, 1, 1]
y_pred = [0, 1, 1, 1, 1, 0, 0, 0, 1, 1]


    

In [3]:

    

cm = confusion_matrix(y_true, y_pred)


    

In [4]:

    

print(cm)


    

    




[[1 4]
 [3 2]]

In [5]:

    

sns.heatmap(cm)
plt.savefig('data/dst/sklearn_confusion_matrix.png')
plt.close()


    

In [6]:

    

sns.heatmap(cm, annot=True, cmap='Blues')
plt.savefig('data/dst/sklearn_confusion_matrix_annot_blues.png')
plt.close()