Tensorflow.js 从下载的文件创建模型

Question

我有一个经过训练的模型，用 const saveResult = await model.save('localstorage://my-model-1'); 保存了它。

现在我想重新加载它并再次使用它。所以我想做这样的事情：

async function loadModel() {
    let myModel = tf.sequential();
    myModel = await model.save('downloads://my-model-1')
    console.log(myModel);

    let outputs = myModel.predict([
        tf.tensor2d([[0, 0, 1]])
    ]);
    outputs.print();
}

但是好像model.savereturns一个modelArtifactsInfo。那么我如何从这个对象创建模型呢？

我关注了 this tutorial 但他们并没有真正解释这一点。

这可能吗？

Answer 1

在教程中说明了一切：

您只需使用您的保存句柄调用 tf.loadModel，其中 returns 一个解析为加载模型或抛出错误的承诺。在您的情况下，句柄将是 'localstorage://my-model-1'.

const model = await tf.loadModel('localstorage://my-model-1');

Tensorflow.js 从下载的文件创建模型

Tensorflow.js create model from downloaded file

tensorflow.js