rixwew/pytorch-fm
PyTorch implementation of factorization machines and neural CTR prediction models including DeepFM, DCN, xDeepFM, and Neural Collaborative Filtering.

Velocity · 7d
+0.4
★ / day
Trend
→steady
star history
This repository provides PyTorch implementations of various factorization machine models and deep learning architectures used for click-through rate prediction and recommendation systems. It includes classical models like FM and FFM alongside deep models such as Wide&Deep, DeepFM, DCN, xDeepFM, and Neural Collaborative Filtering. The package also bundles common benchmark datasets for CTR prediction research.