FM的一个小BUG - Githubissues

ShomyLiu / Neu-Review-Rec

A Toolkit for Neural Review-based Recommendation models with Pytorch.

http://shomy.top/2019/12/31/neu-review-rec/

168 stars 54 forks source link

Closed JarenceSJ closed 4 years ago

JarenceSJ commented 4 years ago

framework/prediciton.py line 136 应该指定sum的dim=1。

ShomyLiu commented 4 years ago

谢谢指出，已经修正:

fm_output = 0.5 * torch.sum(fm_interactions_1 - fm_interactions_2, 1, keepdim=True) + fm_linear_part