TensorflowJs extract model.predict() result in a Javascript variable

šŸ“– Neural networks for Javascript developers

The Neural Networks for JavaScript developers book is almost ready! Learn the basics of AI with TensorFlowJs examples. Join now the presale and get a 15$ coupon off the launching price!

After we configure and train our TensorflowJs model, we can run the predict() function to get the results of that model:

//1. create and configure the TensorflowJs model

//2. train the model

//3. use the model
model.predict(inputs)

If we want to store the results of model.predict() in a separate variable so that we can later refer to it we can do that using the arraySync() function:

const result = model.predict(tf.tensor1d([input_val_here]));
const predictedVal = resultTensor.arraySync()[0];

Keep in mind that, for example, if we use a 2d tensor we will need to update the index as needed:

const result = model.predict(tf.tensor2d([input_val_here], [1, 1]));
const predictedVal = resultTensor.arraySync()[0][0];

Also, you can use the print() function to directly log the result:

model.predict(tf.tensor1d([input_val_here])).print()

šŸ“– Neural networks for Javascript developers

The Neural Networks for JavaScript developers book is almost ready! Learn the basics of AI with TensorFlowJs examples. Join now the presale and get a 15$ coupon off the launching price!


Leave a Reply

Your email address will not be published. Required fields are marked *

Home Screencasts Best of Newsletter Search X

Neural Networks for JavaScript developers
Presale - 15$ free coupon

Hi friend! Before you go, just wanted to let you know that in March 2023 I will be launching the TensorFlow.Js by Example course.

This course will include basic concepts of AI and Neural Networks done with TensorFlowJs such as:

  • Working with datasets
  • Visualizing training
  • Deep Neural Networks in JavaScript
  • Reinforcement Learning
  • Convolutional neural networks
  • and much more ...

Also, there will be a lot of examples as:

  • Object detection
  • Natural language processing
  • Face and voice recognition
  • Gaming AI

Join now the waiting list and get a 15$ coupon off the launching price!

X