This package provides a PyTorch implementation of factorization machine models and common datasets in CTR prediction.
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