xiuxiuxiua 2022-08-09 20:08 采纳率: 50%
浏览 36
已结题

pytorch训练的DeepFM模型的调用

我想要达到的结果

我用pytorch训练了一个DeepFM模型的pth文件该怎么去调用它来实现输入一个用户来推荐相关的物品

  • 写回答

1条回答 默认 最新

  • kakaccys 2022-08-09 20:52
    关注

    没太明白，既然你都训练出来了，不应该知道怎么去推理吗？我理解deepfm输入基本上都是一些one-hot好的特征，你可以直接load训练好的pth文件，然后model(feature)就能输出某一用户点该物品的概率了

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论

报告相同问题？

问题事件

  • 系统已结题 8月18日
  • 已采纳回答 8月10日
  • 创建了问题 8月9日