Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Tutorial on warmstart grid search for multiple classifiers #35

Open
wants to merge 3 commits into
base: master
Choose a base branch
from

Conversation

cshih14
Copy link

@cshih14 cshih14 commented Mar 30, 2020

Solves issue #34

This tutorial tunes hyperparameters max_features and n_estimators for ExtraTrees and RandomForestClassifier in order to fairly compare their performance.

Included in Functions

  • Given classifiers and dictionaries of two hyperparameters, finds the optimal pair of hyperparameter values for each classifier using grid search. In this specific example, max_features and n_estimators will be the parameters being tuned.

  • Implements warmstart for searching through increasing numbers of estimators in the ensemble.

  • Visualizes the performance of each parameter value pair in a heatmap to show more detail on how the optimal parameters were determined.

@cshih14 cshih14 requested a review from j1c March 30, 2020 18:48
@j1c
Copy link
Member

j1c commented Apr 6, 2020

The notebook looks great! I think the last thing to change is to turn the ipynb file into a .py file. In sklearn, the .py files are used to build the examples page automatically, and they wont take ipynb files.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants