Merge pull request #2160 from recommenders-team/miguel/mind_test

Added extra MIND urls
recommenders-team · Aug 30, 2024 · 4bad91f · 4bad91f
2 parents d61fcd5 + 610e663
commit 4bad91f
Showing 1 changed file with 21 additions and 10 deletions.
diff --git a/recommenders/datasets/mind.py b/recommenders/datasets/mind.py
@@ -17,26 +17,37 @@
 )
 
 
-URL_MIND_LARGE_TRAIN = (
-    "https://recodatasets.z20.web.core.windows.net/newsrec/MINDlarge_train.zip"
+URL_MIND_DEMO_TRAIN = (
+    "https://recodatasets.z20.web.core.windows.net/newsrec/MINDdemo_train.zip"
 )
-URL_MIND_LARGE_VALID = (
-    "https://recodatasets.z20.web.core.windows.net/newsrec/MINDlarge_dev.zip"
+URL_MIND_DEMO_VALID = (
+    "https://recodatasets.z20.web.core.windows.net/newsrec/MINDdemo_dev.zip"
+)
+URL_MIND_DEMO_UTILS = (
+    "https://recodatasets.z20.web.core.windows.net/newsrec/MINDdemo_utils.zip"
 )
+
 URL_MIND_SMALL_TRAIN = (
     "https://recodatasets.z20.web.core.windows.net/newsrec/MINDsmall_train.zip"
 )
 URL_MIND_SMALL_VALID = (
     "https://recodatasets.z20.web.core.windows.net/newsrec/MINDsmall_dev.zip"
 )
-URL_MIND_DEMO_TRAIN = (
-    "https://recodatasets.z20.web.core.windows.net/newsrec/MINDdemo_train.zip"
+URL_MIND_SMALL_UTILS = (
+    "https://recodatasets.z20.web.core.windows.net/newsrec/MINDsmall_utils.zip"
 )
-URL_MIND_DEMO_VALID = (
-    "https://recodatasets.z20.web.core.windows.net/newsrec/MINDdemo_dev.zip"
+
+URL_MIND_LARGE_TRAIN = (
+    "https://recodatasets.z20.web.core.windows.net/newsrec/MINDlarge_train.zip"
 )
-URL_MIND_DEMO_UTILS = (
-    "https://recodatasets.z20.web.core.windows.net/newsrec/MINDdemo_utils.zip"
+URL_MIND_LARGE_VALID = (
+    "https://recodatasets.z20.web.core.windows.net/newsrec/MINDlarge_dev.zip"
+)
+URL_MIND_LARGE_TEST = (
+    "https://recodatasets.z20.web.core.windows.net/newsrec/MINDlarge_test.zip"
+)
+URL_MIND_LARGE_UTILS = (
+    "https://recodatasets.z20.web.core.windows.net/newsrec/MINDlarge_utils.zip"
 )
 
 URL_MIND = {