如何在 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]