This package provides a PyTorch implementation of factorization machine models and common datasets in CTR prediction.
- MovieLens Dataset
- Criteo Display Advertising Challenge
- Avazu Click-Through Rate Prediction
- Frappè Dataset
Each model's AUC values are about 0.80 for criteo dataset, and about 0.78 for avazu dataset. (please see example code)
pip install torchfm
https://rixwew.github.io/pytorch-fm
MIT