jeffheaton / t81_558_deep_learning

T81-558: Keras - Applications of Deep Neural Networks @Washington University in St. Louis
https://sites.wustl.edu/jeffheaton/t81-558/
Other
5.72k stars 3.03k forks source link

Is there any way to save the model as .pkl file? #94

Closed kirkegross closed 3 years ago

kirkegross commented 3 years ago

Hi!

Thank you for this code. I'm new to all this. So might be a stupid question, but is it possible to save the model as .pkl instead of .h5 file? Or is there any way to convert it to .pkl file after saving it as .h5 file?

Best, Kirke

jeffheaton commented 3 years ago

I usually save to H5, but more info on pickle here: https://medium.com/@maxwellflitton/pickling-keras-models-c10f7c1b99b1