Skip to content

Commit caef573

Browse files
authored
Merge pull request scikit-learn-contrib#33 from tagomatech/tagomatech-patch-2
changed value of parameter in the example code
2 parents 916b88b + ace7ec0 commit caef573

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)