plt.figure() plot_confusion_matrix(confusion_matrix(rets, pred_rets), classes=class_names, title='Confusion matrix')