Skip to content

Commit 972489e

Browse files
committed
setup e5 study
1 parent d6ad0d1 commit 972489e

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

src/main.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,9 +24,9 @@
2424
METRIC = "loss" # Options: "loss", "ndcg"
2525
STUDY_SET = "full"
2626
CLASSIFIER_TYPE = "svm" # Options: "nb", "log", "svm", "rf"
27-
FEATURE_EXTRACTOR_TYPE = "tfidf" # Options: "tfidf", "onehot", "labse", "bge-m3", "stella", "mxbai", "gist", "e5", "gte", "kalm", "lajavaness", "snowflake"
27+
FEATURE_EXTRACTOR_TYPE = "e5" # Options: "tfidf", "onehot", "labse", "bge-m3", "stella", "mxbai", "gist", "e5", "gte", "kalm", "lajavaness", "snowflake"
2828
PICKLE_FOLDER_PATH = Path("synergy-dataset", f"pickles_{FEATURE_EXTRACTOR_TYPE}")
29-
PRE_PROCESSED_FMS = False # False = on the fly
29+
PRE_PROCESSED_FMS = True # False = on the fly
3030
PARALLELIZE_OBJECTIVE = True
3131
AUTO_SHUTDOWN = False
3232

0 commit comments

Comments
 (0)