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