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

时间:2019-07-19 15:07:18

标签: javascript tensorflow tensorflow.js supervised-learning

我正在尝试将预测系统构建到MERN应用中。我是Tensorflow的新手,并且刚按照this教程学习了每加仑马力和英里数之间的关系。

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]));  

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

请帮助

1 个答案:

答案 0 :(得分:0)

鉴于模型,特征是二维数据。

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

找到其预测值
const tensor = model.predict(tf.tensor([valueOfInput], [1, 1]))
// Get the value
const value = tensor.dataSync()[0]