PyTorch implementation of Caser recommendation model (research paper)
Top 95.9% on SourcePulse
This repository provides a PyTorch implementation of the Convolutional Sequence Embedding Recommendation Model (Caser), a model designed for personalized top-N sequential recommendation. It is intended for researchers and practitioners in the field of recommender systems who need a PyTorch-based solution for sequential recommendation tasks.
How It Works
Caser leverages convolutional neural networks to capture sequential patterns in user behavior. It embeds items into a latent space and then applies convolutional filters to extract features from sequences of these embeddings. These features are then combined with user embeddings and passed through fully-connected layers to predict the next item a user is likely to interact with. This approach allows Caser to model complex, non-linear relationships within user interaction sequences.
Quick Start & Requirements
pip install -r requirements.txt
.train.txt
and test.txt
, with each line containing user item rating
triplets ordered by time. Ratings are converted to 1.Highlighted Details
Maintenance & Community
Licensing & Compatibility
Limitations & Caveats
The project relies on PyTorch v0.4+, which is an older version and may have compatibility issues with current PyTorch releases. The README does not specify a license, which could be a blocker for commercial adoption.
4 years ago
1 day