Factorization Machine models in PyTorch
This package provides a PyTorch implementation of factorization machine models and common datasets in CTR prediction.
Available Datasets
Available Models
Each model's AUC values are about 0.80 for criteo dataset, and about 0.78 for avazu dataset. (please see example code)
Installation
pip install torchfm
API Documentation
https://rixwew.github.io/pytorch-fm
Licence
MIT