Как обновить TensorFlow. js сохраненная / обученная модель - PullRequest
Новая
       19

Как обновить TensorFlow. js сохраненная / обученная модель

1 голос
/ 27 февраля 2020

Привет, я хотел бы знать, как обновить сохраненную модель TensorFlow js в node js,

Я создал нейронную сеть c в TensorFlow js, используя NodeJS, и сохранил обученная модель и она произвела модель. json и файл weights.bin.

Я видел, что мы можем обновить обученную модель в кератах в Python.

Но есть нет примера для обновления модели в 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]))
})

1 Ответ

1 голос
/ 27 февраля 2020

Сохраненная модель может быть загружена обратно и использована для дальнейшего обучения. 

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

Переобученную модель можно сохранить обратно, и это обновит ее вес. Это как если бы модель была обучена как на исходных данных, так и на текущих данных, что означает, что она может предсказать данные, поступающие из обоих источников.

Однако, если новые данные значительно отличаются от исходных данных, модель будет не в состоянии предсказать хорошо из исходных данных снова. Этот вопрос хорошо обсуждается здесь

...