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

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()

📖 50 Javascript, React and NextJs Projects

Learn by doing with this FREE ebook! Not sure what to build? Dive in with 50 projects with project briefs and wireframes! Choose from 8 project categories and get started right away.

📖 50 Javascript, React and NextJs Projects

Learn by doing with this FREE ebook! Not sure what to build? Dive in with 50 projects with project briefs and wireframes! Choose from 8 project categories and get started right away.


Leave a Reply

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

Home Screencasts Best of Newsletter Search X

📖 50 Javascript, React and NextJs Projects

Hi friend! Before you go, just wanted to let you know about the 50 Javascript, React and NextJs Projects FREE ebook.

One of the best ways to learn is by doing the work. Choose from 8 project categories and get started right away:

  • Business & Real-World
  • Games & Puzzles
  • Fun & Interesting
  • Tools & Libraries
  • Personal & Portfolio
  • Project Add-Ons
  • Productivity
  • Clones

Learn by doing with this FREE ebook! Not sure what to build? Dive in with 50 projects complete with project briefs and wireframes!

Keep building and level up! Get all projects as an ebook right to your inbox!

X