text: import matplotlib.pyplot as plt     
  
  # Save a confusion matrix as png
  def plot_cm(cm, labels, filename):
      fig = plt.figure()
      ax = fig.add_subplot(111)
      cax = ax.matshow(cm)
      plt.title('Confusion matrix of the classifier')
      fig.colorbar(cax)
      ax.set_xticklabels([''] + labels)
      ax.set_yticklabels([''] + labels)
      plt.xlabel('Predicted')
      plt.ylabel('True')
      plt.savefig(filename)
   