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

Posted 2023-03-12

【中文标题】如何在 Tensorflow.js 中获取预测值

【英文标题】：How to get predicted value in Tensorflow.js

【发布时间】：2019-11-28 14:39:59

【问题描述】：

我正在尝试将预测系统构建到 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】：

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

如果你有一个输入字段值，你可以通过使用找到它的预测

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

【讨论】：

我在上面添加了一行，控制台记录了它。以下是响应 - "isDisposedInternal":false,"shape":[1,1],"dtype":"float32","size":1,"strides":[1],"dataId": ,"id":10892,"rankType":"2"。是不是应该这样。我期待输入的单个值

const price = model.predict(tf.tensor([96087], [1, 1]));

如果要获取值，可以使用datasync()或data()。你可以看到我编辑的答案。