如何更新 TensorFlow.js 保存/训练的模型

Question

您好，我想知道如何在 node js 中更新 TensorFlow js 保存的模型，

我使用 NodeJS 在 TensorFlow js 中创建了一个基本的神经网络，并保存了训练好的模型，并生成了 model.json 和 weights.bin 文件。

我已经看到我们可以在 Python 中更新 keras 中训练好的模型。

但是TensorFlow js、NodeJS和JS中没有更新模型的例子。

请有人帮忙更新训练好的模型。

模型加载和更新代码

var tf = require('@tensorflow/tfjs-node')
async funtion load(){
    const model = await tf.loadLayersModel(url);
}
load().then(() => {
    model.predict(tf.tensor1d([5]))
    model.predict(tf.tensor1d([5]), tf.tensor1d([15]));
    model.predict(tf.tensor1d([5]))
})

Answer 1

可以加载保存的模型并用于进一步训练。

const model = await tf.loadLayersModel(url);
model.predict(feature) // predict with old model
await model.fit(features, labels); // this will update the weights of the model
model.predict(feature) // predict with new model

可以保存重新训练的模型，这将更新其权重。就好像模型在初始数据和当前数据上都进行了训练，这意味着它可以预测来自这两个来源的数据。

但是，如果新数据与初始数据存在显着差异，则模型将无法再次根据初始数据进行良好预测。这个问题好好讨论here