Pytorch_DeepFM

Pytorch_DeepFM copied to clipboard

兄弟,你俩谁抄袭的谁?

1

这里有一个一模一样的: https://github.com/chenxijun1029/DeepFM_with_PyTorch

htmars-hub

bias未设置device,导致use_cuda=True时报错 RuntimeError: expected device cuda:0 **but** got device cpu 可以修改为： bias = torch.nn.Parameter(torch.randn(Xi.size(0))).to(self.device)

NIck-Meng

请问大神Xv全1矩阵的作用是什么

linduchina

您好， 请问下原文中在 deep part中有使用 激活函数， 而这份代码中没用，是出于其他什么考虑吗？

duan348733684

这不是份完整的代码？

2

感谢分享，读完代码后感觉这不是一份完整的代码，FM层应该是不完整的，直接使用了Embedding做计算，latent矩阵没有做是吗？

heyxhh