This is the official implementation for the "Adaptive Session-based Recommendation with Contrastive Learning" paper. It includes data processing of datasets (RetailRocket and Diginetica) that we used to train and test our model.
- CONTRAST Data Pipeline: Clean, transform, and structure raw data into usable session datasets.
- CONTRAST Model Implementation: Architecture with training, evaluation, and visualization.
The framework was evaluated on two benchmark datasets:
Dataset | Train Sessions | Test Sessions | Unique Items | Avg. Session Length |
---|---|---|---|---|
RetailRocket | 433,643 | 15,132 | 41,762 | 5.64 |
Diginetica | 719,470 | 60,858 | 37,867 | 4.06 |
The preprocessing steps are encapsulated in the Data Pipeline Notebook.
To experiment with the CONTRAST model or its data pipeline:
- Open the corresponding Kaggle notebook links.
- Fork the notebook to your Kaggle account.
- Run all cells (ensure GPU is enabled for faster training).
Licensed under the Apache License.