Skip to content

Navigation Menu

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

Appearance settings

cnfsll/Neural-Factorization-Machine

Repository files navigation

Neural-Factorization-Machine

基于TensorFlow实现Neural-Factorization-Machine
参考如下:
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.
https://github.com/hexiangnan/neural_factorization_machine

LoadData.py:数据读取
NeuralFM_Model.py:模型定义
Run_NeuralFM_SquareLoss.py:针对平方误差损失,训练模型
Run_NeuralFM_LogLoss.py:针对对数似然损失,训练模型(对于frappe数据集,采用该损失很难找到合适的超参数)

About

Factorization Machine, Deep Learning, Recommender System

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

Morty Proxy This is a proxified and sanitized view of the page, visit original site.