test saga solver

asreview · Jan 23, 2025 · f597599 · f597599
1 parent 6651a2f
commit f597599
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/asreview2-optuna/classifiers.py b/asreview2-optuna/classifiers.py
@@ -18,7 +18,7 @@ def logistic_params(trial: optuna.trial.FrozenTrial):
     # Use logarithmic normal distribution for C (C effect is non-linear)
     C = trial.suggest_float("log__C", 0.01, 10, log=True)
 
-    solver = "lbfgs"
+    solver = trial.suggest_categorical("log__solver", ["lbfgs", "saga"])
 
     return {"C": C, "solver": solver}
 

diff --git a/asreview2-optuna/main.py b/asreview2-optuna/main.py
@@ -21,7 +21,7 @@
 from feature_extractors import feature_extractor_params, feature_extractors
 
 # Study variables
-VERSION = 1
+VERSION = 2
 STUDY_SET = "demo"
 PICKLE_FOLDER_PATH = Path("synergy-dataset", "pickles")
 CLASSIFIER_TYPE = "log"  # Options: "nb", "log", "svm", "rf"