SPORF icon indicating copy to clipboard operation
SPORF copied to clipboard

add additional sklearn parameters

Open falkben opened this issue 5 years ago • 1 comments

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
  • [x] oob_score
  • [ ] verbose
  • [ ] warm_start
  • [ ] class_weight

Some of these may be of interest to support. Some are relatively easy to add.

falkben avatar Apr 12 '19 20:04 falkben

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.

bkj avatar Jul 12 '19 20:07 bkj