recommenders-team/recommenders

[ASK] Save NCF Model

Open

#1 735 ouverte le 8 juin 2022

Voir sur GitHub
 (8 commentaires) (0 réactions) (0 assignés)Python (2 972 forks)batch import
help wanted

Métriques du dépôt

Stars
 (17 706 stars)
Métriques de merge PR
 (Merge moyen 6j 16h) (10 PRs mergées en 30 j)

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

Guide contributeur

Direction de recherche
L'utilisateur souhaite enregistrer le modèle NCF pour l'inférence sans nécessiter le code original de construction du modèle. Les exemples du dépôt ne montrent que l'enregistrement pour un entraînement ultérieur. Pour résoudre ce problème, étudiez la classe du modèle NCF (probablement dans ncf.py) et identifiez les méthodes de sérialisation. Envisagez d'utiliser torch.save(model.state dict(), ...) de PyTorch ou model.save() de TensorFlow. De plus, vérifiez si le dépôt dispose d'utilitaires ou de scripts d'exportation pour le service de modèle. Consultez le notebook deep dive pour le contexte et toute issue ou discussion connexe (par exemple, les commentaires de l'issue #1735) pouvant fournir des indices. Proposez d'ajouter une section de documentation ou un notebook d'exemple démontrant comment enregistrer et charger le modèle pour l'inférence.
Stack technique
python
Domaine
machine learning
Type d'issue
Recherche
DifficultéDifficulté estimée pour un nouveau contributeur, de 1 pour un très petit changement à 5 pour un travail expert.
2
Temps estiméFourchette de temps approximative pour investiguer, implémenter, tester et préparer une pull request.
1-3 heures
Statut d'activitéDisponibilité apparente de l'issue : fraîche, active, ancienne, bloquée ou en attente d'un mainteneur.
Active
ClartéClarté avec laquelle l'issue explique le changement attendu, les critères d'acceptation et la prochaine étape.
Claire
Prérequis
Python
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
75