Skip to content

Commit 3ef5b53

Browse files
committed
setup doc2vec
1 parent ea1dcee commit 3ef5b53

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 = "demo"
2626
CLASSIFIER_TYPE = "svm" # Options: "nb", "log", "svm", "rf"
27-
FEATURE_EXTRACTOR_TYPE = "e5" # Options: "tfidf", "onehot", "labse", "bge-m3", "stella", "mxbai", "gist", "e5", "gte", "kalm", "lajavaness", "snowflake"
27+
FEATURE_EXTRACTOR_TYPE = "doc2vec" # 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)