hexiangnan / Neural_factorization_machine
TenforFlow Implementation of Neural Factorization Machine
Stars: ✭ 422
Programming Languages
python
139335 projects - #7 most used programming language
Projects that are alternatives of or similar to Neural factorization machine
Rankfm
Factorization Machines for Recommendation and Ranking Problems with Implicit Feedback Data
Stars: ✭ 71 (-83.18%)
Mutual labels: recommender-system, factorization-machines
Attentional Neural Factorization Machine
Attention,Factorization Machine, Deep Learning, Recommender System
Stars: ✭ 39 (-90.76%)
Mutual labels: recommender-system, 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 (-33.65%)
Mutual labels: recommender-system, factorization-machines
Openlearning4deeprecsys
Some deep learning based recsys for open learning.
Stars: ✭ 383 (-9.24%)
Mutual labels: recommender-system, factorization-machines
Rsparse
Fast and accurate machine learning on sparse matrices - matrix factorizations, regression, classification, top-N recommendations.
Stars: ✭ 145 (-65.64%)
Mutual labels: recommender-system, factorization-machines
Flurs
🌊 FluRS: A Python library for streaming recommendation algorithms
Stars: ✭ 97 (-77.01%)
Mutual labels: recommender-system, factorization-machines
Fastfm
fastFM: A Library for Factorization Machines
Stars: ✭ 908 (+115.17%)
Mutual labels: recommender-system, factorization-machines
Recommendation.jl
Building recommender systems in Julia
Stars: ✭ 42 (-90.05%)
Mutual labels: recommender-system, factorization-machines
Attentional factorization machine
TensorFlow Implementation of Attentional Factorization Machine
Stars: ✭ 362 (-14.22%)
Mutual labels: recommender-system, factorization-machines
Summary Of Recommender System Papers
阅读过的推荐系统论文的归类总结，持续更新中…
Stars: ✭ 288 (-31.75%)
Mutual labels: recommender-system
Dsin
Code for the IJCAI'19 paper "Deep Session Interest Network for Click-Through Rate Prediction"
Stars: ✭ 289 (-31.52%)
Mutual labels: recommender-system
Rspapers
A Curated List of Must-read Papers on Recommender System.
Stars: ✭ 4,140 (+881.04%)
Mutual labels: recommender-system
Artificio
Deep Learning Computer Vision Algorithms for Real-World Use
Stars: ✭ 326 (-22.75%)
Mutual labels: recommender-system
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 (+603.32%)
Mutual labels: factorization-machines
Recnn
Reinforced Recommendation toolkit built around pytorch 1.7
Stars: ✭ 362 (-14.22%)
Mutual labels: recommender-system
Tensorflow Xnn
Tensorflow implementation of DeepFM variant that won 4th Place in Mercari Price Suggestion Challenge on Kaggle.
Stars: ✭ 263 (-37.68%)
Mutual labels: factorization-machines
ds3-spring-2018
Материалы третьего набора офлайн-программы Data Scientist.
Stars: ✭ 22 (-94.79%)
Mutual labels: recommender-system
Sasrec
SASRec: Self-Attentive Sequential Recommendation
Stars: ✭ 373 (-11.61%)
Mutual labels: recommender-system
Lightfm
A Python implementation of LightFM, a hybrid recommendation algorithm.
Stars: ✭ 3,884 (+820.38%)
Mutual labels: recommender-system
Neural Factorization Machines
This is our implementation for the paper:
Xiangnan He and Tat-Seng Chua (2017). Neural Factorization Machines for Sparse Predictive Analytics. In Proceedings of SIGIR '17, Shinjuku, Tokyo, Japan, August 07-11, 2017.
We have additionally released our TensorFlow implementation of Factorization Machines under our proposed neural network framework.
Please cite our SIGIR'17 paper if you use our codes. Thanks!
Author: Dr. Xiangnan He (http://www.comp.nus.edu.sg/~xiangnan/)
Example to run the codes.
python NeuralFM.py --dataset frappe --hidden_factor 64 --layers [64] --keep_prob [0.8,0.5] --loss_type square_loss --activation relu --pretrain 0 --optimizer AdagradOptimizer --lr 0.05 --batch_norm 1 --verbose 1 --early_stop 1 --epoch 200
The instruction of commands has been clearly stated in the codes (see the parse_args function).
The current implementation supports two tasks: regression and binary classification. The regression task optimizes RMSE, and the binary classification task optimizes Log Loss.
Dataset
We use the same input format as the LibFM toolkit (http://www.libfm.org/).
Split the data to train/test/validation files to run the codes directly (examples see data/frappe/).
Last Update Date: May 11, 2017
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].