2016-07-24 1 views
0

Je souhaite former mon propre modèle Word2Vec pour mon corpus de texte. Je peux obtenir le code du tutoriel de TensorFlow. Ce que je ne sais pas est comment sauvegarder ce modèle à utiliser pour la classification de texte CNN plus tard? Devrais-je utiliser du cornichon pour le sauvegarder et le lire plus tard?Enregistrement de Word2Vec pour CNN Classification textuelle

Répondre

0

Vous pouvez utiliser pickle pour l'enregistrer sur le disque. Ensuite, lorsque vous créez le modèle CNN, chargez la table d'incorporation de mots enregistrée et utilisez-la pour initialiser la variable TensorFlow qui contient les incorporations de mots pour votre classificateur CNN.

0

Pas de décapage n'est pas le moyen de sauver le modèle en cas de flux tensoriel.

Tensorflow fournit un service tensorflow permettant d'enregistrer les modèles en tant que proto-bufs (pour l'exportation du modèle). La façon de sauver le modèle serait de sauver la session de tensorflow comme: saver.save (sess, 'my_test_model', global_step = 1000)

Heres le lien pour réponse complète: Tensorflow: how to save/restore a model?