feat: use balanced_subsample

tokebe · Jun 7, 2022 · bffd50e · bffd50e
1 parent 0cd90c7
commit bffd50e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/niclassify/core/classifiers.py b/niclassify/core/classifiers.py
@@ -459,7 +459,7 @@ def __init__(self):
         """
         super().__init__(
             clf=RandomForestClassifier(
-                class_weight="balanced", oob_score=True, n_estimators=1000
+                class_weight="balanced_subsample", oob_score=True, n_estimators=1000
             ),
             hyperparams={
                 'max_depth': list(np.linspace(10, 50).astype(int)),