search

tensorflow / skflow

Simplified interface for TensorFlow (mimicking Scikit Learn) for Deep Learning
Apache License 2.0
3.18k stars 439 forks source link

Text classification - How to save VocabularyProcessor after fit? #130

Closed prasanna513 closed 8 years ago

prasanna513 commented 8 years ago

To later run the saved model on test data, I need to save the VocabularyProcessor after I fit with the training data. How do I save it? Thanks, Prasanna

terrytangyuan commented 8 years ago

You can dump it to JSON file. @ilblackdragon Do you think it's worth making those serializable or implementing serializer method?

prasanna513 commented 8 years ago

Save and restore methods on this too will be helpful I think. Thanks for the quick reply.

ilblackdragon commented 8 years ago

You can just pickle the object. I'll add save and restore methods though.