Add next-item pipeline (#561)

* Add `data.dataset.SequentialDataset` to support new data formats: * "SIT": Session, Item, Timestamp * "SITJson": Session, Item, Timestamp, Json * "USIT": User, Session, Item, Timestamp * "USITJson": User, Session, Item, Timestamp, Json * Add `data.reader` parser for listed formats * Add `eval_method.NextItemEvaluation` for next-item evaluation * Add `models.SPop` baseline model: session-based popular * Add `gowalla` check-ins and `yoochoose` datasets * Add unit tests --------- Co-authored-by: lthoang <[email protected]> Co-authored-by: tqtg <[email protected]>
PreferredAI · Dec 26, 2023 · eabdd2d · eabdd2d
1 parent 477405a
commit eabdd2d
Show file tree

Hide file tree

Showing 18 changed files with 1,401 additions and 79 deletions.
diff --git a/cornac/data/__init__.py b/cornac/data/__init__.py
@@ -22,13 +22,17 @@
 from .reader import Reader
 from .dataset import Dataset
 from .dataset import BasketDataset
+from .dataset import SequentialDataset
 
-__all__ = ['FeatureModality',
-           'TextModality',
-           'ReviewModality',
-           'ImageModality',
-           'GraphModality',
-           'SentimentModality',
-           'BasketDataset',
-           'Dataset',
-           'Reader']
+__all__ = [
+    "FeatureModality",
+    "TextModality",
+    "ReviewModality",
+    "ImageModality",
+    "GraphModality",
+    "SentimentModality",
+    "BasketDataset",
+    "SequentialDataset",
+    "Dataset",
+    "Reader",
+]