changed value of parameter in the 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

changed value of parameter in the example code #33

Closed tagomatech closed 6 years ago

tagomatech commented 6 years ago

Replaced 'auto' with 'balanced' on line 168, as 'auto' is an invalid value parameter rf = RandomForestClassifier(n_jobs=-1, class_weight='balanced', max_depth=5)

danielhomola commented 6 years ago

thanks!