rixwew / pytorch-fm

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

why don't you do the pre-trianing in FNN model? #25

Open BrandonCXY opened 4 years ago

BrandonCXY commented 4 years ago

按照论文,FNN模型(即 FactorizationSupportedNeuralNetworkModel)应该是利用了FM模型pre-traing后的隐向量作为MLP的输入,但是我在您的FactorizationSupportedNeuralNetworkModel的code中发现并没有实现这一点。而且FNN model(即FactorizationSupportedNeuralNetworkModel) 的实现方式与您写的Wide&Deep model 的Deep部分的实现方式一模一样,这应该是不对。 希望大佬可以探讨一下,或许是我理解不到位,或者是代码实现有问题,谢谢!(抱歉,这部分拿英文复述我怕自己说得不明确,因此用中文)