Skip to content

关于NFM #17

@TQCAI

Description

@TQCAI

博主,您可以理解错了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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions