setup bge-m3

asreview · Jan 28, 2025 · 635348e · 635348e
1 parent 22b20ad
commit 635348e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/asreview2-optuna/main.py b/asreview2-optuna/main.py
@@ -22,7 +22,7 @@
 VERSION = 1
 STUDY_SET = "demo"
 CLASSIFIER_TYPE = "svm"  # Options: "nb", "log", "svm", "rf"
-FEATURE_EXTRACTOR_TYPE = "labse"  # Options: "tfidf", "onehot", "labse", "bge-m3"
+FEATURE_EXTRACTOR_TYPE = "bge-m3"  # Options: "tfidf", "onehot", "labse", "bge-m3"
 PICKLE_FOLDER_PATH = Path("synergy-dataset", f"pickles_{FEATURE_EXTRACTOR_TYPE}")
 PRE_PROCESSED_FMS = True  # False = on the fly
 PARALLELIZE_OBJECTIVE = True