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.

Sign up for GitHub

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

Jump to bottom

add additional sklearn parameters #238

Open
1 of 10 tasks
falkben opened this issue Apr 12, 2019 · 1 comment
Open
1 of 10 tasks

add additional sklearn parameters #238

falkben opened this issue Apr 12, 2019 · 1 comment
Labels
C++ Python

Comments

@falkben
Copy link
Contributor

falkben commented Apr 12, 2019 •
edited
Loading

The following is a list of parameters that sklearn random forest has that we do not.

  • criterion (gini/entropy)
  • min_samples_leaf: The minimum number of samples required to be at a leaf node.
  • min_weight_fraction_leaf
  • max_leaf_nodes
  • min_impurity_decrease
  • bootstrap
  • oob_score
  • verbose
  • warm_start
  • class_weight

Some of these may be of interest to support. Some are relatively easy to add.
@bkj
Copy link

bkj commented Jul 12, 2019

I'm trying to run some larger-scale hyperparameter tuning experiments --warm_start would be particularly useful.

A verbose flag would also be very nice to keep logs clean.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
C++ Python
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@falkben @bkj