DeepFM调参总结
生活随笔
收集整理的這篇文章主要介紹了
DeepFM调参总结
小編覺得挺不錯(cuò)的,現(xiàn)在分享給大家,幫大家做個(gè)參考.
一、結(jié)論
二、應(yīng)用場(chǎng)景
2.1 指標(biāo)選擇
應(yīng)用deepfm和lr模型的目的是給召回階段的候選集合排序,故選擇AUC為模型離線評(píng)測(cè)的指標(biāo)。同時(shí)關(guān)注loss的變化情況。
2.2 數(shù)據(jù)情況
三、實(shí)驗(yàn)數(shù)據(jù)
默認(rèn)參數(shù):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 |
結(jié)合訓(xùn)練時(shí)間,性能等實(shí)際情況,最優(yōu)等參數(shù)如下:
特征: uid,itemid
embedding:3
dnn結(jié)構(gòu):(128,128,128)
dropout:0.85
四、實(shí)驗(yàn)思考
https://arxiv.org/pdf/2006.06863.pdf
總結(jié)
以上是生活随笔為你收集整理的DeepFM调参总结的全部?jī)?nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: Linux命令:ls、grep、wc统计
- 下一篇: 日常生活小技巧 -- 玩转 PDF