本程序是利用Pytorch实现DeepFM算法,以Criteo数据集作为测试,AUC值目前可以达到79.15%
模型实现细节见: https://blog.csdn.net/springtostring/article/details/108157070
参考代码库DeepCTR:https://github.com/shenweichen/DeepCTR-Torch
本程序是利用Pytorch实现DeepFM算法,以Criteo数据集作为测试,AUC值目前可以达到79.15%
模型实现细节见: https://blog.csdn.net/springtostring/article/details/108157070
参考代码库DeepCTR:https://github.com/shenweichen/DeepCTR-Torch