setup mxbai study

asreview · Feb 5, 2025 · 5dc268f · 5dc268f
1 parent a65d435
commit 5dc268f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/asreview2-optuna/main.py b/asreview2-optuna/main.py
@@ -21,7 +21,7 @@
 # Study variables
 VERSION = 1
 METRIC = "ndcg"  # Options: "loss", "ndcg"
-STUDY_SET = "full"
+STUDY_SET = "demo"
 CLASSIFIER_TYPE = "xgboost"  # Options: "nb", "log", "svm", "rf", "xgboost"
 FEATURE_EXTRACTOR_TYPE = "mxbai"  # Options: "tfidf", "onehot", "labse", "bge-m3", "stella", "mxbai"
 PICKLE_FOLDER_PATH = Path("synergy-dataset", f"pickles_{FEATURE_EXTRACTOR_TYPE}")
@@ -248,7 +248,7 @@ def download_pickles(report_order):
 
 if __name__ == "__main__":
     # list of studies
-    studies = pd.read_json(f"synergy_studies_{STUDY_SET}.jsonl", lines=True).head(1)
+    studies = pd.read_json(f"synergy_studies_{STUDY_SET}.jsonl", lines=True)
     report_order = sorted(set(studies["dataset_id"]))
 
     if PRE_PROCESSED_FMS: