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.save 返回了 modelArtifactsInfo。那么如何从这个对象创建模型呢？

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

这可能吗？

Answer 1

在教程中它说明了一切：

您只需要使用您的保存句柄调用tf.loadModel，它会返回一个解析为已加载模型或引发错误的承诺。在您的情况下，句柄将是 'localstorage://my-model-1'。

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