From 2f982b23e865f81588cce45e157efff1a7f7c7f3 Mon Sep 17 00:00:00 2001 From: timovdk <5330531+timovdk@users.noreply.github.com> Date: Tue, 28 Jan 2025 16:48:22 +0100 Subject: [PATCH] setup labse --- 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 295c235..e0214de 100644 --- a/asreview2-optuna/main.py +++ b/asreview2-optuna/main.py @@ -22,7 +22,7 @@ VERSION = 1 STUDY_SET = "demo" CLASSIFIER_TYPE = "rf" # Options: "nb", "log", "svm", "rf" -FEATURE_EXTRACTOR_TYPE = "bge-m3" # Options: "tfidf", "onehot", "labse", "bge-m3" +FEATURE_EXTRACTOR_TYPE = "labse" # 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