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.