caser_pytorch  by graytowne

PyTorch implementation of Caser recommendation model (research paper)

created 7 years ago
271 stars

Top 95.9% on SourcePulse

GitHubView on GitHub
Project Summary

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

  • Install with pip install -r requirements.txt.
  • Requires Python 2 or 3, PyTorch v0.4+, NumPy, and SciPy.
  • Data should be formatted as train.txt and test.txt, with each line containing user item rating triplets ordered by time. Ratings are converted to 1.

Highlighted Details

  • Implements the Caser model from the WSDM '18 paper.
  • Claims potential for better performance than reported in the original paper, with mAP reaching 0.17 on a test set with specific configurations (d=50, L=5, T=3).
  • Heavily built on the Spotlight library for utilities and interactions.

Maintenance & Community

  • No explicit information on maintainers, community channels, or roadmap is provided in the README.

Licensing & Compatibility

  • The README does not specify a license. Compatibility for commercial use or closed-source linking is not addressed.

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.

Health Check
Last commit

4 years ago

Responsiveness

1 day

Pull Requests (30d)
0
Issues (30d)
0
Star History
2 stars in the last 90 days

Explore Similar Projects

Starred by Chip Huyen Chip Huyen(Author of AI Engineering, Designing Machine Learning Systems), Didier Lopes Didier Lopes(Founder of OpenBB), and
11 more.

sentence-transformers by UKPLab

0.2%
17k
Framework for text embeddings, retrieval, and reranking
created 6 years ago
updated 6 days ago
Feedback? Help us improve.