javascript - 如何保存 Tensorflow.js 模型？

Question

我想制作一个用户界面来创建、保存和训练 tensorflow.js 模型。但是我无法在创建模型后保存模型。我什至从 tensorflow.js 文档中复制了这段代码，但它不起作用:

const model = tf.sequential(
     {layers: [tf.layers.dense({units: 1, inputShape: [3]})]});
console.log('Prediction from original model:');
model.predict(tf.ones([1, 3])).print();

const saveResults = await model.save('localstorage://my-model-1');

const loadedModel = await tf.loadModel('localstorage://my-model-1');
console.log('Prediction from loaded model:');
loadedModel.predict(tf.ones([1, 3])).print();

我总是收到错误消息“Uncaught SyntaxError: await is only valid in async function”。我该如何解决这个问题？谢谢!

Answer 1

您需要处于异步环境中。创建一个异步函数(async function name(){...})并在需要时调用它，或者最短的方法是自调用异步箭头函数:

(async ()=>{
   //you can use await in here
})()