Keras函数API在每个API中更改层名称

2024-06-11 18:09:02 发布

您现在位置:Python中文网/ 问答频道 /正文

当我在k-fold交叉验证模型中运行函数API时,在每个fold的回归拟合模型中,命名稠密层的数量会增加。 就像在第一次折叠中它是dense_2_acc,然后在第二次折叠中它是dense_5_acc。在

由我的模型总结表明我的模型是正确的。为什么要更改每个折叠的拟合模型history对象的名称?在

enter image description here


Tags: 对象函数模型名称api数量fold命名
1条回答
网友
1楼 · 发布于 2024-06-11 18:09:02

这是一个非常好的问题,它展示了keras的一些重要内容。名称以这种方式更改的原因是keras没有清除以前定义的变量,即使覆盖模型时也是如此。您可以通过调用以下命令轻松检查变量是否仍在session.graph中:

from keras import backend as K
K.get_session().graph.get_collection('variables')

为了清除之前的模型变量,可以调用:

^{pr2}$

但是,请小心,因为您可能会丢失现有的模型。如果您想保持名称不变,只需在层实例化中添加name参数来命名层,例如:

^{3}$

相关问题 更多 >