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

问题描述 投票:4回答:1

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

python tensorflow keras jupyter-notebook
1个回答
11
投票

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

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

© www.soinside.com 2019 - 2024. All rights reserved.