Setup labse

asreview · Jan 28, 2025 · aab5428 · aab5428
1 parent 82a72a3
commit aab5428
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/asreview2-optuna/classifiers.py b/asreview2-optuna/classifiers.py
@@ -2,7 +2,6 @@
 from asreview.models.classifiers import (
     LogisticClassifier,
     NaiveBayesClassifier,
-    RandomForestClassifier,
     SVMClassifier,
 )
 
@@ -41,7 +40,9 @@ def random_forest_params(trial: optuna.trial.FrozenTrial):
     n_estimators = trial.suggest_int("rf__n_estimators", 50, 200)
 
     # Use normal distribution for max_features (max_features effect is linear)
-    max_features = trial.suggest_categorical("rf__max_features", ["sqrt", "log2"])
+    max_features = (
+        None  # trial.suggest_categorical("rf__max_features", ["sqrt", "log2"])
+    )
     return {"n_estimators": n_estimators, "max_features": max_features}
 
 
@@ -70,6 +71,7 @@ def __init__(self, n_estimators=100, max_features=10, **kwargs):
             **kwargs,
         )
 
+
 classifiers = {
     "nb": NaiveBayesClassifier,
     "log": LogisticClassifier,