setup mxbai

asreview · Feb 7, 2025 · 0fc2daf · 0fc2daf
1 parent e6f0c66
commit 0fc2daf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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