From a511ddb4c71d903edfbe773198ee41cc0d49ca23 Mon Sep 17 00:00:00 2001
From: timovdk <5330531+timovdk@users.noreply.github.com>
Date: Tue, 28 Jan 2025 16:47:40 +0100
Subject: [PATCH] switch to demo mode

---
 asreview2-optuna/classifiers.py | 5 ++---
 asreview2-optuna/main.py        | 2 +-
 2 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/asreview2-optuna/classifiers.py b/asreview2-optuna/classifiers.py
index 23adb1f..1ea1c7a 100644
--- a/asreview2-optuna/classifiers.py
+++ b/asreview2-optuna/classifiers.py
@@ -40,9 +40,8 @@ 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 = (
-        None  # trial.suggest_categorical("rf__max_features", ["sqrt", "log2"])
-    )
+    max_features = trial.suggest_categorical("rf__max_features", ["sqrt", "log2"])
+
     return {"n_estimators": n_estimators, "max_features": max_features}
 
 
diff --git a/asreview2-optuna/main.py b/asreview2-optuna/main.py
index 0c95cb4..295c235 100644
--- a/asreview2-optuna/main.py
+++ b/asreview2-optuna/main.py
@@ -20,7 +20,7 @@
 
 # Study variables
 VERSION = 1
-STUDY_SET = "full"
+STUDY_SET = "demo"
 CLASSIFIER_TYPE = "rf"  # Options: "nb", "log", "svm", "rf"
 FEATURE_EXTRACTOR_TYPE = "bge-m3"  # Options: "tfidf", "onehot", "labse", "bge-m3"
 PICKLE_FOLDER_PATH = Path("synergy-dataset", f"pickles_{FEATURE_EXTRACTOR_TYPE}")