Skip to content

Commit 07bfbac

Browse files
authored
replaced parameter value in example code
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)
1 parent 64a41d2 commit 07bfbac

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

boruta/boruta_py.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,7 @@ class BorutaPy(BaseEstimator, TransformerMixin):
150150
151151
# define random forest classifier, with utilising all cores and
152152
# sampling in proportion to y labels
153-
rf = RandomForestClassifier(n_jobs=-1, class_weight='auto', max_depth=5)
153+
rf = RandomForestClassifier(n_jobs=-1, class_weight='balanced', max_depth=5)
154154
155155
# define Boruta feature selection method
156156
feat_selector = BorutaPy(rf, n_estimators='auto', verbose=2, random_state=1)

0 commit comments

Comments
 (0)