Open TQCAI opened 3 years ago
博主，您可以理解错了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
博主，您可以理解错了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