将 TensorFlow JS model.predict 的值获取到变量中

Question

我正在尝试在 JS 中实现一个神经网络，它可以预测某些值，然后将这些值存储在变量中以供以后使用。

async function processModel(inputs) {
  const model = await tf.loadLayersModel(modelURL);
  inputs = tf.tensor(inputs);

  var predictions = model.predict(inputs);
  predictions = predictions.dataSync();

  console.log(typeof(predictions));

  return predictions;
}

启动它告诉我这predictions是类型object。我想获取返回的 Promise 值并将这些值存储在变量中，以便我可以将它们与其他变量进行比较。

有人可以帮我解决这个问题吗？

Answer 1

prredictions在一个数组中。数组索引是获取预测值的方式