用Google搜索“TensorFlow+DeepFM”,一般都能搜索到“ChenglongChen/tensorflow-DeepFM”和“lambdaJi的TensorFlow Estimator of DeepFM”这二位的实现。二位不仅用TensorFlow实现了DeepFM,还在Criteo数据集上,给出了完整的训练、测试的代码,的确给了我很大的启发,在这里要表示感谢。
deepfm :
- embedding 层共享,fm和deep共享embeding层,fm训练得到的参数作为wide部分的输出,也作为dnn部分的输入
- 端到端训练,embedding和网络权重联合训练,无需预训练和单独训练
https://www.zhihu.com/search?type=content&q=deepfm%E4%BB%A3%E7%A0%81
DeepFM算法说明
http://blog.sina.com.cn/s/blog_628cc2b70102yooc.html
{
"passenger_count": {
"1.0": 0,
"2.0": 1,
"3.0": 2
},
"hitch_rate": 3,
"pax1_rating_section": {
"1": 4,
"2": 5,
"3": 6,
"4": 7,
"5": 8
}
}
Xi_test[0] [0,3,8]
Xi_test[1] [2,3,5]
Xv_test[0] [1.0, 1.3360856379927106, 1.0]
Xv_test[1] [1.0, 0.0346312255213426, 1.0]
深入FFM原理与实践
https://tech.meituan.com/2016/03/03/deep-understanding-of-ffm-principles-and-practices.html
七月在线 ctr预估视频
http://www.julyedu.com/video/play/42/2172###
知乎讲解以及代码wiki
https://zhuanlan.zhihu.com/p/35815532
深度学习技巧之Batch Normalization
https://www.datalearner.com/blog/1051550556076499