Text classification - How to save VocabularyProcessor after fit?
prasanna513 opened this issue · comments
Prasanna Kumar commented
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
Yuan Tang commented
You can dump it to JSON file. @ilblackdragon Do you think it's worth making those serializable or implementing serializer method?
Prasanna Kumar commented
Save and restore methods on this too will be helpful I think. Thanks for the quick reply.
Illia Polosukhin commented
You can just pickle the object. I'll add save and restore methods though.