Skip to content

Commit 0245491

Browse files
committed
setup e5
1 parent 5b41c26 commit 0245491

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@
2424
METRIC = "loss" # Options: "loss", "ndcg"
2525
STUDY_SET = "full"
2626
CLASSIFIER_TYPE = "svm" # Options: "nb", "log", "svm", "rf"
27-
FEATURE_EXTRACTOR_TYPE = "mxbai" # 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}")
2929
PRE_PROCESSED_FMS = True # False = on the fly
3030
PARALLELIZE_OBJECTIVE = True

0 commit comments

Comments
 (0)