Skip to content

Instantly share code, notes, and snippets.

@drorhilman
Created November 17, 2017 10:16
Show Gist options
  • Save drorhilman/0da6e5cbdd3a8f0ea627036dc3f49c20 to your computer and use it in GitHub Desktop.
Save drorhilman/0da6e5cbdd3a8f0ea627036dc3f49c20 to your computer and use it in GitHub Desktop.
from sklearn.metrics import confusion_matrix
import seaborn as sns
forest.fit(x,y)
predict = forest.predict(x)
cm = pd.DataFrame(confusion_matrix(y, predict), columns=forest.classes_, index=forest.classes_)
sns.heatmap(cm)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment