recommenders-team/recommenders

[ASK] Save NCF Model

Open

#1.735 aberto em 8 de jun. de 2022

Ver no GitHub
 (8 comments) (0 reactions) (0 assignees)Python (2.972 forks)batch import
help wanted

Métricas do repositório

Stars
 (17.706 stars)
Métricas de merge de PR
 (Mesclagem média 6d 16h) (10 fundiu PRs em 30d)

Description

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

Guia do colaborador

Direção de pesquisa
O usuário deseja salvar o modelo NCF para inferência sem exigir o código original de construção do modelo. Os exemplos no repositório mostram apenas o salvamento para treinamento adicional. Para resolver isso, investigue a classe do modelo NCF (provavelmente em ncf.py) e identifique métodos de serialização. Considere usar torch.save(model.state dict(), ...) do PyTorch ou model.save() do TensorFlow. Além disso, verifique se o repositório possui utilitários ou scripts de exportação para servir o modelo. Examine o notebook deep dive para contexto e quaisquer issues ou discussões relacionadas (por exemplo, comentários do issue #1735) que possam fornecer pistas. Proponha adicionar uma seção de documentação ou um notebook de exemplo demonstrando como salvar e carregar o modelo para inferência.
Pilha de tecnologia
python
Domain
machine learning
Tipo Issue
Pesquisa
DifficultyDificuldade de implementação estimada para um novo contribuidor, de 1 para alterações muito pequenas a 5 para trabalho de nível especializado.
2
Tempo estimadoUm intervalo de tempo aproximado para um colaborador experiente investigar, implementar, testar e preparar um pull request.
1-3 horas
Status da atividadeQuão disponível o issue aparece agora: novo, ativo, obsoleto, bloqueado ou aguardando entrada do mantenedor.
Ativo
ClarityCom que clareza o issue explica a mudança esperada, os critérios de aceitação e a próxima etapa.
Claro
Prerequisites
Python
Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.
75