From 0fc2dafecf98a740893be5695827ba48ec2ee59e Mon Sep 17 00:00:00 2001
From: timovdk <5330531+timovdk@users.noreply.github.com>
Date: Fri, 7 Feb 2025 13:52:22 +0100
Subject: [PATCH] setup mxbai

---
 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 f121c87..0e7622e 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 = "svm"  # Options: "nb", "log", "svm", "rf"
-FEATURE_EXTRACTOR_TYPE = "stella"  # Options: "tfidf", "onehot", "labse", "bge-m3", "stella", "mxbai"
+FEATURE_EXTRACTOR_TYPE = "mxbai"  # 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