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

Question

嗨，我想知道如何更新节点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

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

然而，如果新数据与初始数据有显著差异，则模型将无法再次根据初始数据进行良好预测。这个问题在这里讨论得很好