Skip to content

Instantly share code, notes, and snippets.

@xoelop

xoelop/heatmaps_random_forest.py

Last active November 3, 2018 18:56
Show Gist options
  • Save xoelop/a332473d897df4dacc0a023ec55c0111 to your computer and use it in GitHub Desktop.
Save xoelop/a332473d897df4dacc0a023ec55c0111 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
heatmaps_random_forest.py
max_scores = df_gridsearch.groupby(['param_min_samples_split',
'param_max_features']).max()
max_scores = max_scores.unstack()[['mean_test_score', 'mean_train_score']]
sns.heatmap(max_scores.mean_test_score, annot=True, fmt='.4g');
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment