← all repositories

rixwew/pytorch-fm

PyTorch implementation of factorization machines and neural CTR prediction models including DeepFM, DCN, xDeepFM, and Neural Collaborative Filtering.

1.1k stars Python ML Frameworks
pytorch-fm
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.

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