简体   繁体   中英

How to get predicted value in Tensorflow.js

I am trying build predictive system into a MERN app. I am new to Tensorflow and just followed this tutorial which gets relation between Horsepower and Miles per gallon.

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

I've not understood how to utilize this to get a predicted value. How do I enter a horsepower in input field and get predicted MPG.

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

The above line is used to plot range of horsepower from low to high I guess. But how do I get a predicted value for entered value?

Please help

Given the model, the features are a 2d dimensional data.

If you have an input field value, you can find its prediction by using

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

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM