在Keras中重置模型的所有权重和偏置(训练后恢复模型)

假设我有如下代码。

model = Sequential()model.add(LSTM(units = 10 input_shape = (x1, x2)))model.add(Activation('tanh'))model.compile(optimizer = 'adam', loss = 'mse')## Step 1.model.fit(X_train, Y_train, epochs = 10)

在训练模型之后,我想重置模型中的所有内容(权重和偏置)。所以我想在compile函数之后(步骤1)恢复模型。在Keras中最快的实现方式是什么?


回答:

是否是最快的方法可能还有待商榷,但这种方法确实简单直观,可能对你的情况已经足够了:序列化初始权重,然后在需要时反序列化,并使用io.BytesIO来避免磁盘I/O操作(以及后续的清理工作):

from io import BytesIOmodel = Sequential()model.add(LSTM(units = 10, input_shape = (x1, x2)))model.add(Activation('tanh'))model.compile(optimizer = 'adam', loss = 'mse')f = BytesIO()model.save_weights(f)  # 存储权重model.fit(X_train, Y_train, epochs = 10)# [在这里对训练后的模型做任何你想做的事情]model.load_weights(f)  # 重置权重

Related Posts

在使用k近邻算法时,有没有办法获取被使用的“邻居”?

我想找到一种方法来确定在我的knn算法中实际使用了哪些…

Theano在Google Colab上无法启用GPU支持

我在尝试使用Theano库训练一个模型。由于我的电脑内…

准确性评分似乎有误

这里是代码: from sklearn.metrics…

Keras Functional API: “错误检查输入时:期望input_1具有4个维度,但得到形状为(X, Y)的数组”

我在尝试使用Keras的fit_generator来训…

如何使用sklearn.datasets.make_classification在指定范围内生成合成数据?

我想为分类问题创建合成数据。我使用了sklearn.d…

如何处理预测时不在训练集中的标签

已关闭。 此问题与编程或软件开发无关。目前不接受回答。…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注