Skip to content

Instantly share code, notes, and snippets.

@VictorSaenger
Last active September 7, 2019 14:32
Show Gist options
  • Select an option

  • Save VictorSaenger/2a1d8daff3360c13d06c4dd0340cae54 to your computer and use it in GitHub Desktop.

Select an option

Save VictorSaenger/2a1d8daff3360c13d06c4dd0340cae54 to your computer and use it in GitHub Desktop.
plt.figure(figsize=(4, 3))
sns.heatmap(confusion_matrix(class_test, class_pred), \
annot=True,linewidths=2, cmap="YlGnBu",fmt="g", \
xticklabels=['Romanticism','Realism','Surrealism'],\
yticklabels=['Romanticism','Realism','Surrealism'])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment