← all repositories

maciejkula/spotlight

PyTorch-based library for building deep and shallow recommender models with factorization and sequence models.

3k stars Python Domain AppsML Frameworks
spotlight
Velocity · 7d
+0.9
★ / day
Trend
steady
star history

Spotlight is a Python library that leverages PyTorch to construct recommender system models. It provides building blocks for loss functions including pointwise and pairwise ranking losses, representations such as shallow factorization and deep sequence models, and utilities for fetching or generating recommendation datasets. The library aims to facilitate rapid exploration and prototyping of new recommendation models.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.