rixwew / pytorch-fm

Factorization Machine models in PyTorch
MIT License
1.03k stars 225 forks source link

xDeepFM #1

Open hecc1231 opened 5 years ago

hecc1231 commented 5 years ago

关于CIN的结构,是否在卷积层前将二三维度调换一下?

rixwew commented 5 years ago

Splitting hidden vectors would improve CIN performance please see the author's github issue https://github.com/Leavingseason/xDeepFM/issues/5