CODEWITHSUNDEEP
javascripttensorflowtensorflow.jssupervised-learning
Stacy J
Stacy J

Reputation: 2781

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

Upvotes: 3

Views: 4969

Answers (1)

edkeveked
edkeveked

Reputation: 18381

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]

Upvotes: 5

Related Questions