2016-07-24 6 views
0

Ich möchte mein eigenes Word2Vec-Modell für meinen Textkorpus trainieren. Ich kann den Code von TensorFlow Tutorial bekommen. Was ich nicht weiß ist, wie man dieses Modell speichert, um es später für die CNN-Textklassifizierung zu verwenden? Sollte ich Essiggurke verwenden, um es zu speichern und es später zu lesen?Word2Vec für CNN Text-Klassifizierung speichern

Antwort

0

Sie können Pickle verwenden, um es auf der Festplatte zu speichern. Laden Sie dann beim Erstellen des CNN-Modells die gespeicherte Worteinbettungstabelle und verwenden Sie diese, um die TensorFlow-Variable zu initialisieren, die die Worteinbettungen für Ihren CNN-Klassifizierer enthält.

0

Kein Beizen ist nicht die Art, das Modell bei Tensorflow zu speichern.

Tensorflow bietet mit Tensorflow zum Speichern der Modelle als Proto bufs (zum Exportieren des Modells). Die Art und Weise Modell zu retten wäre, die tensorflow Sitzung speichern als: saver.save (sess, 'my_test_model', global_step = 1000)

Heres den Link für eine vollständige Antwort: Tensorflow: how to save/restore a model?