CODEWITHSUNDEEP
pythontensorflow
Domen P
Domen P

Reputation: 173

Saving wide and deep tensorflow model

I'm playing with Wide and Deep learning example from Tensorflow. I would like to save the trained classifier to be used for prediction tasks later on but I don't really see how. The DNNLinearCombinedClassifier doesn't have a save method available and pickling the object fails as well.

Any ideas how to save it?

Upvotes: 2

Views: 814

Answers (2)

Alexandre Passos
Alexandre Passos

Reputation: 5206

The checkpoint saving section of this doc should answer your question.

Upvotes: 2

Jacob Holloway
Jacob Holloway

Reputation: 887

Are you looking for https://www.tensorflow.org/versions/r0.10/how_tos/variables/index.html#saving-and-restoring?

Specifically Saving Variables and Restoring Variables. Additionally, you can use the Checkpoint Files to save the weights periodically (useful if your computer stops training halfway through, you don't have to start from the beginning).

Upvotes: 0

Related Questions