replaced parameter value in example code

scikit-learn-contrib / boruta_py

Python implementations of the Boruta all-relevant feature selection method.

BSD 3-Clause "New" or "Revised" License

1.46k stars 252 forks source link

replaced parameter value in example code #34

Closed tagomatech closed 6 years ago

tagomatech commented 6 years ago

Replaced parameter value 'auto' in example code with 'balanced' which is a valid value rf = RandomForestClassifier(n_jobs=-1, class_weight='balanced', max_depth=5)

danielhomola commented 6 years ago

thanks! this must have changed, as in older scikit it was definitely 'auto'