Factorization Machine models in PyTorch

Name: rixwew/pytorch-fm
Brand: rixwew/pytorch-fm
SKU: project/rixwew/pytorch-fm
Rating: 4.52 (268 reviews)

This package provides a PyTorch implementation of factorization machine models and common datasets in CTR prediction.

Available Datasets

Available Models

Model	Reference
Logistic Regression
Factorization Machine	S Rendle, Factorization Machines, 2010.
Field-aware Factorization Machine	Y Juan, et al. Field-aware Factorization Machines for CTR Prediction, 2015.
Higher-Order Factorization Machines	M Blondel, et al. Higher-Order Factorization Machines, 2016.
Factorization-Supported Neural Network	W Zhang, et al. Deep Learning over Multi-field Categorical Data - A Case Study on User Response Prediction, 2016.
Wide&Deep	HT Cheng, et al. Wide & Deep Learning for Recommender Systems, 2016.
Attentional Factorization Machine	J Xiao, et al. Attentional Factorization Machines: Learning the Weight of Feature Interactions via Attention Networks, 2017.
Neural Factorization Machine	X He and TS Chua, Neural Factorization Machines for Sparse Predictive Analytics, 2017.
Neural Collaborative Filtering	X He, et al. Neural Collaborative Filtering, 2017.
Field-aware Neural Factorization Machine	L Zhang, et al. Field-aware Neural Factorization Machine for Click-Through Rate Prediction, 2019.
Product Neural Network	Y Qu, et al. Product-based Neural Networks for User Response Prediction, 2016.
Deep Cross Network	R Wang, et al. Deep & Cross Network for Ad Click Predictions, 2017.
DeepFM	H Guo, et al. DeepFM: A Factorization-Machine based Neural Network for CTR Prediction, 2017.
xDeepFM	J Lian, et al. xDeepFM: Combining Explicit and Implicit Feature Interactions for Recommender Systems, 2018.
AutoInt (Automatic Feature Interaction Model)	W Song, et al. AutoInt: Automatic Feature Interaction Learning via Self-Attentive Neural Networks, 2018.
AFN(AdaptiveFactorizationNetwork Model)	Cheng W, et al. Adaptive Factorization Network: Learning Adaptive-Order Feature Interactions, AAAI'20.

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

Project Name	Stars	Most Recent Commit	Total Releases	Latest Release	Open Issues	License	Language
Daisyrec	496	2 months ago	15	August 14, 2022		mit	Python
This is the repository of our article published in RecSys 2020 "Are We Evaluating Rigorously? Benchmarking Recommendation for Reproducible Evaluation and Fair Comparison" and of several follow-up studies.
Pytorch Fm	268	3 years ago	8	January 18, 2020	3	mit	Python
Factorization Machine models in PyTorch
Recstudio	140	10 days ago	11	April 16, 2022	6	mit	Python
A highly-modularized and recommendation-efficient recommendation library based on PyTorch.
Recommendation_model	116	a year ago			4		Python
练习下用pytorch来复现下经典的推荐系统模型, 如MF, FM, DeepConn, MMOE, PLE, DeepFM, NFM, DCN, AFM, AutoInt, ONN, FiBiNET, DCN-v2, AFN, DCAP等
Factorization Machine For Prediction	88	6 years ago			1	apache-2.0	Jupyter Notebook
Factorization Machine for regression and classification
Nfm Pyorch	12	4 years ago					Python
A pytorch implementation for He et al. Neural Factorization Machines for Sparse Predictive Analytics on SIGIR 2017.
Deeprec Torch	8	3 years ago				mit	Python
Easy-to-use pytorch-based framework for RecSys models
Torchctr	6	4 years ago	2	September 02, 2019			Python
CTR Prediction on PyTorch
Fm For Online Recommendation	4	a year ago					Jupyter Notebook
Factorization machine model and its variations for recommendation systems
Attentive Factorization Machine	2	4 years ago					Python
Pytorch_AFM

Pytorch Fm