gpt4 book ai didi

python - 如何 pickle Keras 模型?

转载 作者:太空狗 更新时间:2023-10-29 18:02:44 24 4
gpt4 key购买 nike

官方文档说“不推荐使用 pickle 或 cPickle 来保存 Keras 模型。”

但是,我对 pickle Keras 模型的需求源于使用 sklearn 的 RandomizedSearchCV(或任何其他超参数优化器)的超参数优化。必须将结果保存到文件中,因为这样脚本就可以在分离的 session 等中远程执行。

本质上,我想:

trial_search = RandomizedSearchCV( estimator=keras_model, ... )
pickle.dump( trial_search, open( "trial_search.pickle", "wb" ) )

最佳答案

截至目前,Keras 模型是可 pickle 的。但我们仍然建议使用 model.save() 将模型保存到磁盘。

关于python - 如何 pickle Keras 模型?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/48295661/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com