search

nfmcclure / tensorflow_cookbook

Code for Tensorflow Machine Learning Cookbook
https://www.packtpub.com/big-data-and-business-intelligence/tensorflow-machine-learning-cookbook-second-edition
MIT License
6.23k stars 2.41k forks source link

how to add model persistance to this spam filter using pickle #146

Open vanipolnedi opened 6 years ago

vanipolnedi commented 6 years ago

can anyone please suggest me how to add pickling (model persistance) to this spam filter. thanks in advance :)

nfmcclure commented 6 years ago

Hi @vanipolnedi , Thanks for the request, I think we should be able to add this. Were you set on using pickle? Pickle tends to be very very sensitive to python versions and pickle versions (and object versions). I think it would be better to create some TensorFlow model objects for loading after training. I can get started on that this weekend. Are you set on pickle? I think the TensorFlow saving operation is a bit better.

vanipolnedi commented 6 years ago

@nfmcclure thank you for the reply. it will be very useful for us if u add pickling to the spam filter ( which uses rnn)