Confusion Matrix with Seaborn

confusion_matrix.py

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

cm = confusion_matrix(y_test, y_pred)

plt.figure(figsize=(12,12))

sns.heatmap(data=cm, # Data to plot
            square=True, # Make the cells square, looks nicer.
            cmap='Reds', # Colour scheme to use
            xticklabels=model.classes_, # Labels for axis
            yticklabels=model.classes_, # Labels for axis
            annot=True, # add numbers to cells
            fmt='g'); # format cell numbers as int