nzc / dnn_ctr

The framework to deal with ctr problem。The project contains FNN,PNN,DEEPFM, NFM etc
756 stars 285 forks source link

关于NFM模型 #1

Open benzfy opened 6 years ago

benzfy commented 6 years ago

image 原文里面用了下面的这个式子简化了运算 NFM.py里面好像是用的循环,这样是否会效率较低?

nzc commented 6 years ago

你好,在原论文的bi-interaction中,我的代码是使用了该公式进行加速的。详情可以看NFM.py的195行-202行~ @benzfy

benzfy commented 6 years ago

Sorry,我看到后面的else去了,else分支里面,FM本身也是这样加速的吧。