GitPlanet

Cheap and reliable Node.js hosting starts at $3/month, and $1/month static HTML hosting

Created with love in Canada, visit hostnodejs.com today

Feel like to post an Ad? Learn Details
All Projects → rixwew → Pytorch Fm

rixwew / Pytorch Fm

Licence: mit
Factorization Machine models in PyTorch

Programming Languages

python
139335 projects - #7 most used programming language

Labels

pytorchcollaborative-filteringfactorization-machines

Projects that are alternatives of or similar to Pytorch Fm

Rankfm
Factorization Machines for Recommendation and Ranking Problems with Implicit Feedback Data
Stars: ✭ 71 (-84.4%)
Mutual labels:  collaborative-filtering, factorization-machines
Daisyrec
A developing recommender system in pytorch. Algorithm: KNN, LFM, SLIM, NeuMF, FM, DeepFM, VAE and so on, which aims to fair comparison for recommender system benchmarks
Stars: ✭ 280 (-38.46%)
Mutual labels:  collaborative-filtering, factorization-machines
Rsparse
Fast and accurate machine learning on sparse matrices - matrix factorizations, regression, classification, top-N recommendations.
Stars: ✭ 145 (-68.13%)
Mutual labels:  collaborative-filtering, factorization-machines
Deepmatch
A deep matching model library for recommendations & advertising. It's easy to train models and to export representation vectors which can be used for ANN search.
Stars: ✭ 1,051 (+130.99%)
Mutual labels:  collaborative-filtering, factorization-machines
Recommendation.jl
Building recommender systems in Julia
Stars: ✭ 42 (-90.77%)
Mutual labels:  collaborative-filtering, factorization-machines
Tensorflow Xnn
Tensorflow implementation of DeepFM variant that won 4th Place in Mercari Price Suggestion Challenge on Kaggle.
Stars: ✭ 263 (-42.2%)
Mutual labels:  factorization-machines
Cornac
A Comparative Framework for Multimodal Recommender Systems
Stars: ✭ 308 (-32.31%)
Mutual labels:  collaborative-filtering
recommender
NReco Recommender is a .NET port of Apache Mahout CF java engine (standalone, non-Hadoop version)
Stars: ✭ 35 (-92.31%)
Mutual labels:  collaborative-filtering
Movie Recommendation Engine
Movie Recommender based on the MovieLens Dataset (ml-100k) using item-item collaborative filtering.
Stars: ✭ 21 (-95.38%)
Mutual labels:  collaborative-filtering
Neural factorization machine
TenforFlow Implementation of Neural Factorization Machine
Stars: ✭ 422 (-7.25%)
Mutual labels:  factorization-machines
Attentional factorization machine
TensorFlow Implementation of Attentional Factorization Machine
Stars: ✭ 362 (-20.44%)
Mutual labels:  factorization-machines
Summary Of Recommender System Papers
阅读过的推荐系统论文的归类总结，持续更新中…
Stars: ✭ 288 (-36.7%)
Mutual labels:  collaborative-filtering
Neural Collaborative Filtering
pytorch version of neural collaborative filtering
Stars: ✭ 263 (-42.2%)
Mutual labels:  collaborative-filtering
Recommendation Systems Paperlist
Papers about recommendation systems that I am interested in
Stars: ✭ 308 (-32.31%)
Mutual labels:  collaborative-filtering
Spotify-Song-Recommendation-ML
UC Berkeley team's submission for RecSys Challenge 2018
Stars: ✭ 70 (-84.62%)
Mutual labels:  collaborative-filtering
Openlearning4deeprecsys
Some deep learning based recsys for open learning.
Stars: ✭ 383 (-15.82%)
Mutual labels:  factorization-machines
recsys spark
Spark SQL 实现 ItemCF，UserCF，Swing，推荐系统，推荐算法，协同过滤
Stars: ✭ 76 (-83.3%)
Mutual labels:  collaborative-filtering
Ytk Learn
Ytk-learn is a distributed machine learning library which implements most of popular machine learning algorithms(GBDT, GBRT, Mixture Logistic Regression, Gradient Boosting Soft Tree, Factorization Machines, Field-aware Factorization Machines, Logistic Regression, Softmax).
Stars: ✭ 337 (-25.93%)
Mutual labels:  factorization-machines
Librec
LibRec: A Leading Java Library for Recommender Systems, see
Stars: ✭ 3,045 (+569.23%)
Mutual labels:  collaborative-filtering
Xlearn
High performance, easy-to-use, and scalable machine learning (ML) package, including linear model (LR), factorization machines (FM), and field-aware factorization machines (FFM) for Python and CLI interface.
Stars: ✭ 2,968 (+552.31%)
Mutual labels:  factorization-machines
View All Similar Projects ➔

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

Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].