Skip to content

Commit b5ab61f

Browse files
authored
Merge pull request scikit-learn-contrib#34 from tagomatech/tagomatech-patch-3
replaced parameter value in example code
2 parents caef573 + 07bfbac commit b5ab61f

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)