javascript - 将 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;
}

启动此功能会通知我预测是对象类型。我想获取返回的 Promise 值并将这些值存储在变量中，以便我可以将它们与其他变量进行比较。

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

Answer 1

您可以使用 arraySync() 方法访问输出张量中的预测值。例如如果输出张量是一个只有单个输出值的二维数组，那么下面的代码将为您提供预测值。

 const resultTensor = model.predict(tf.tensor2d([50], [1, 1]));
 const predictedValue = resultTensor.arraySync()[0][0];
 console.log(predictedValue);