DeepFM调参总结
生活随笔
收集整理的這篇文章主要介紹了
DeepFM调参总结
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
一、結論
二、應用場景
2.1 指標選擇
應用deepfm和lr模型的目的是給召回階段的候選集合排序,故選擇AUC為模型離線評測的指標。同時關注loss的變化情況。
2.2 數據情況
三、實驗數據
默認參數:epoch=1;optimizer=adam;loss=cross entropy;batch=256;
| 1 | uid,itemid | 10 | dnn=(128,128) | 0 | AUC=0.731 | loss=1.18 |
| 2 | uid,itemid | 2 | dnn=(128,128) | 0 | AUC=0.733 | loss=1.12 |
| 3 | uid,itemid | 4 | dnn=(128,128) | 0 | AUC=0.731 | loss=1.08 |
| 4 | uid,itemid | 3 | dnn=(128,128) | 0.85 | AUC=0.746 | loss=1.17 |
| 5 | uid,itemid | 3 | dnn=(128,128) | 0.65 | AUC=0.736 | loss=1.11 |
| 6 | uid,itemid | 3 | dnn=(256,256) | 0.85 | AUC=0.744 | loss=1.07 |
| 7 | uid,itemid | 3 | dnn=(64,64) | 0.85 | AUC=0.745 | loss=1.24 |
| 8 | uid,itemid | 3 | dnn=(64,64,64) | 0.85 | AUC=0.770 | loss=0.667 |
| 9 | uid,itemid | 3 | dnn=(128,128,128) | 0.85 | AUC=0.772 | loss=0.800 |
結合訓練時間,性能等實際情況,最優等參數如下:
特征: uid,itemid
embedding:3
dnn結構:(128,128,128)
dropout:0.85
四、實驗思考
https://arxiv.org/pdf/2006.06863.pdf
總結
以上是生活随笔為你收集整理的DeepFM调参总结的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: Linux命令:ls、grep、wc统计
- 下一篇: 日常生活小技巧 -- 玩转 PDF