如何在 Tensorflow.js 中获取预测值

Question

我正在尝试将预测系统构建到MERN应用程序中。我是 Tensorflow 的新手，只是按照本教程了解马力和每加仑英里数之间的关系。

https://codelabs.developers.google.com/codelabs/tfjs-training-regression/index.html#6

我不明白如何利用它来获得预测值。如何在输入字段中输入马力并获得预测的 MPG。

const xs = tf.linspace(0, 1, 100);      
  const preds = model.predict(xs.reshape([100, 1]));  

我猜上面的线用于绘制从低到高的马力范围。但是如何获得输入值的预测值呢？

请帮忙

Answer 1

给定模型，特征是二维数据。

如果您有输入字段值，则可以使用

const tensor = model.predict(tf.tensor([valueOfInput], [1, 1]))
// Get the value
const value = tensor.dataSync()[0]