From 375e5da465bc16fdd8b5fed15618cf38b5dcb685 Mon Sep 17 00:00:00 2001 From: timovdk <5330531+timovdk@users.noreply.github.com> Date: Tue, 28 Jan 2025 16:09:07 +0100 Subject: [PATCH] setup bge-m3 --- 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 5b49d76..0c95cb4 100644 --- a/asreview2-optuna/main.py +++ b/asreview2-optuna/main.py @@ -22,7 +22,7 @@ VERSION = 1 STUDY_SET = "full" CLASSIFIER_TYPE = "rf" # 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