如何清除使用Keras和Tensorflow创建的模型(作为后端)?

Rav*_*rip 11 python keras tensorflow jupyter-notebook

在Jupyter Notebook中使用Keras训练神经网络时遇到问题.我创建了一个包含多个隐藏层的顺序模型.在训练模型并保存结果之后,我想删除此模型并在同一会话中创建一个新模型,因为我有一个for循环来检查不同参数的结果.但是当我理解我得到的错误时,在更改参数时,当我循环时,我只是在模型中添加图层(即使我network = Sequential()在循环内部再次初始化它).所以我的问题是,如何完全清除以前的模型,或者如何在同一个会话中初始化一个全新的模型?

小智 25

keras.backend.clear_session()应该清除以前的型号.来自https://keras.io/backend/:

销毁当前的TF图并创建一个新图.有助于避免旧模型/图层混乱.

  • 如果我只想删除与特定模型关联的图形部分怎么办?keras.backend.clear_session()有点像核选项 (4认同)
  • @Zach,也许你可以运行 `model_i_want.save('model_dest.h5')`、`keras.backend.clear_session()` 然后加载你的模型 `model_i_want = tf.keras.load_model('model_dest.h5')` ? (2认同)