From 1a922e4296d654a7fa37def8c6af11df898824eb Mon Sep 17 00:00:00 2001
From: timovdk <5330531+timovdk@users.noreply.github.com>
Date: Wed, 5 Feb 2025 16:56:05 +0100
Subject: [PATCH] setup stella study

---
 asreview2-optuna/main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/asreview2-optuna/main.py b/asreview2-optuna/main.py
index 84b08fc..0cd0b89 100644
--- a/asreview2-optuna/main.py
+++ b/asreview2-optuna/main.py
@@ -23,7 +23,7 @@
 METRIC = "ndcg"  # Options: "loss", "ndcg"
 STUDY_SET = "demo"
 CLASSIFIER_TYPE = "rf"  # Options: "nb", "log", "svm", "rf"
-FEATURE_EXTRACTOR_TYPE = "mxbai"  # Options: "tfidf", "onehot", "labse", "bge-m3", "stella", "mxbai"
+FEATURE_EXTRACTOR_TYPE = "stella"  # Options: "tfidf", "onehot", "labse", "bge-m3", "stella", "mxbai"
 PICKLE_FOLDER_PATH = Path("synergy-dataset", f"pickles_{FEATURE_EXTRACTOR_TYPE}")
 PRE_PROCESSED_FMS = True  # False = on the fly
 PARALLELIZE_OBJECTIVE = True