Skip to content

Files

Failed to load latest commit information.

Latest commit

 Cannot retrieve latest commit at this time.

History

History

DeepFM4CTR

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 
 
 
 
 
 
 

通过tensorflow实现dnn、fm与deepfm三种ctr预测模型,训练出的model file可通过tensorflow serving用于线上推荐打分。
传统的DNN虽然能够学习到非常复杂的高阶特征,但是对于高度稀疏的特征来说,这种隐式交叉并未带来足够的优异性,有必要引入新的层级结构:1. 设计新的layer,完成显式的特征交叉;2. 借助LR/FM等线性模型,帮助DNN学习更好的特征表示
运行方式:
python deepfm_train4tesla.py --log_dir ./data/test.txt --model_type 0
通过脚本中参数model_type可控制0 deepfm ,1 only fm ,2 only dnn
通过tensorboard观察测试数据在3种模型auc和loss等指标
项目详细介绍见KM分享DeepFM模型-模型实践篇
tensorflow graph
实现方法参考:A Factorization-Machine based Neural Network for CTR Prediction