recommenders-team/recommenders

[ASK] Save NCF Model

Open

#1,735 创建于 2022年6月8日

在 GitHub 查看
 (8 评论) (0 反应) (0 负责人)Python (2,972 fork)batch import
help wanted

仓库指标

Star
 (17,706 star)
PR 合并指标
 (平均合并 6天 16小时) (30 天内合并 10 个 PR)

描述

Description

I want to save the entire NCF model and use it without requiring the original model building code to run. How can I do that? The example in the deep dive notebook only shows how to save the model for further training which is not what I am looking for. Appreciate any reponses.

Other Comments

贡献者指南

研究方向
用户希望保存NCF模型用于推理，而无需依赖原始模型构建代码。仓库中的示例仅展示了为继续训练而保存模型。为解决此问题，需研究NCF模型类（可能在ncf.py中），并识别序列化方法。考虑使用PyTorch的torch.save(model.state dict(), ...)或TensorFlow的model.save()。此外，检查仓库是否包含任何用于模型服务的导出工具或脚本。参考deep dive notebook以获取上下文，并查看相关issue或讨论（例如issue #1735的评论），这些可能提供线索。建议添加文档章节或示例notebook，演示如何保存和加载模型用于推理。
技术栈
python
领域
machine learning
议题类型
调研
难度面向新贡献者的预计实现难度，1 表示很小改动，5 表示专家级工作。
2
预计时间有经验贡献者完成调查、实现、测试并准备 pull request 的粗略时间范围。
1-3 小时
活动状态议题当前的可参与程度：新鲜、活跃、陈旧、阻塞或等待维护者输入。
活跃
清晰度议题是否清楚说明期望改动、验收标准和下一步。
清晰
前置要求
Python
新手友好度1-100 的估计分数，表示该议题对首次贡献者的友好程度。
75