github地址
https://github.com/ChenglongChen/tensorflow-DeepFM
数据集
https://www.kaggle.com/c/porto-seguro-safe-driver-prediction
效果比较
|
deepFM |
FM |
DNN |
normed gini |
0.26685 (0.00119) |
0.26872 (0.00327) |
0.25871 (0.00108) |
gpu耗时/分钟 |
24 |
20 |
19 |
用2核cpu跑了半小时,没有输出任何结果。
用单个2080gpu跑。显存只用了10%,gpu只用了1%,gpu算力没有充分利用起来。