recommenders-team/recommenders

[ASK] Save NCF Model

Open

#1735 aperta il 8 giu 2022

Vedi su GitHub
 (8 commenti) (0 reazioni) (0 assegnatari)Python (2972 fork)batch import
help wanted

Metriche repository

Star
 (17.706 star)
Metriche merge PR
 (Merge medio 6g 16h) (10 PR mergiate in 30 g)

Descrizione

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

Guida contributor

Direzione di ricerca
L'utente desidera salvare il modello NCF per l'inferenza senza richiedere il codice originale di costruzione del modello. Gli esempi nel repository mostrano solo il salvataggio per ulteriore addestramento. Per affrontare questo problema, investigare la classe del modello NCF (probabilmente in ncf.py) e identificare i metodi di serializzazione. Considerare l'uso di torch.save(model.state dict(), ...) di PyTorch o model.save() di TensorFlow. Inoltre, verificare se il repository dispone di utility o script di esportazione per il servizio del modello. Esaminare il notebook deep dive per il contesto e eventuali issue o discussioni correlate (ad esempio, i commenti dell'issue #1735) che potrebbero fornire indizi. Proporre l'aggiunta di una sezione della documentazione o di un notebook di esempio che dimostri come salvare e caricare il modello per l'inferenza.
Tech stack
python
Dominio
machine learning
Tipo issue
Ricerca
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
2
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
1-3 ore
Stato attivitàQuanto è probabile che la issue sia ancora attiva e reviewable.
Attiva
ChiarezzaQuanto chiaramente la issue descrive problema, scope e risultato atteso.
Chiara
Prerequisiti
Python
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
75