New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

关于NFM #17

Open

TQCAI opened this issue Feb 10, 2021 · 0 comments

TQCAI commented Feb 10, 2021

博主，您可以理解错了NFM了

https://github.com/nzc/dnn_ctr/blob/master/model/NFM.py#L235

看到这里，假设数据有39个域，FM的embed向量长度为4，你用FM构造的向量是 39*(39-1)/2 =741 个，为各个域隐向量两两之间的内积。

但是，我看到官方实现的方法是用FM求和后的隐向量，在这个场景下长度应为4

您可以看下官方实现（翻译成torch版的了，原版TF应该和这个差不多）

https://github.com/guoyang9/NFM-pyorch

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment