Skip to content

Commit ace7ec0

Browse files
authored
changed value of parameter in the example code
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)
1 parent 64a41d2 commit ace7ec0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -165,7 +165,7 @@ __verbose__ : int, default=0
165165

166166
# define random forest classifier, with utilising all cores and
167167
# sampling in proportion to y labels
168-
rf = RandomForestClassifier(n_jobs=-1, class_weight='auto', max_depth=5)
168+
rf = RandomForestClassifier(n_jobs=-1, class_weight='balanced', max_depth=5)
169169

170170
# define Boruta feature selection method
171171
feat_selector = BorutaPy(rf, n_estimators='auto', verbose=2, random_state=1)

0 commit comments

Comments
 (0)