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

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.
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

Metadata

Release files for torchfm 0.7.0

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for torchfm 0.7.0
File	Size	Uploaded	Provenance
torchfm-0.7.0.tar.gz	9.6 kB	Jan 18, 2020	Without provenance	Details

Release files / torchfm-0.7.0.tar.gz

Download URL	torchfm-0.7.0.tar.gz
Size	9.6 kB
Tags	Source
SHA-256 checksum How to use checksums	`215838acff337f0abbeaee8e0f3530556a7d524cf35564d0bb6a0d95055a5f52`
BLAKE2b-256 checksum How to use checksums	`ff4b1814b23bed7642d102f0906f7919f0378a05a1b9d2d53c6d5506af20a505`
Upload date	Jan 18, 2020
Uploaded using Trusted Publishing? What is trusted publishing?	No
Uploaded via	`twine/1.14.0 pkginfo/1.5.0.1 requests/2.22.0 setuptools/41.2.0 requests-toolbelt/0.9.1 tqdm/4.35.0 CPython/3.7.3`

Security for torchfm 0.7.0

Report project as malware

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

No provenance found

None of the files in this release were uploaded with provenance metadata.

A note on package security

PyPI is a repository for distributing packages, not a service for auditing them. While third-party security researchers and our staff do their best to detect and remove malicious packages, PyPI cannot guarantee the quality, security or suitability of any given package.

To learn more about protecting yourself from malicious actors, visit our package security documentation.