Skip to content

CONTRAST, a novel architecture designed to enhance session-based recommendation by incorporating memory-efficient sparse operations, attention guided graph convolution, and contrastive learning techniques.

License

Notifications You must be signed in to change notification settings

farneet24/CONTRAST

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 

Repository files navigation

CONTRAST: Contrastive Temperature-Regulated Adaptive Session Training

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.

🔗 Notebooks on Kaggle


Datasets

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.


Performance Table

Performance Table


Running the Notebooks

To experiment with the CONTRAST model or its data pipeline:

  1. Open the corresponding Kaggle notebook links.
  2. Fork the notebook to your Kaggle account.
  3. Run all cells (ensure GPU is enabled for faster training).

License

License: Apache 2.0

Licensed under the Apache License.

About

CONTRAST, a novel architecture designed to enhance session-based recommendation by incorporating memory-efficient sparse operations, attention guided graph convolution, and contrastive learning techniques.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published