tensorflow / skflow

Simplified interface for TensorFlow (mimicking Scikit Learn) for Deep Learning

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Text classification - How to save VocabularyProcessor after fit?

prasanna513 opened this issue · comments

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

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

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

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