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

downloadDownload PNG downloadDownload JPEG downloadDownload SVG

Tip: You can change the style, width & colours of the snippet with the inspect tool before clicking Download!

Click to optimize width for Twitter