Skip to content

Instantly share code, notes, and snippets.

@sararob
Created March 30, 2018 18:02
Show Gist options
  • Star 4 You must be signed in to star a gist
  • Fork 2 You must be signed in to fork a gist
  • Save sararob/f4104d792c65b92dbc2ea31388de6880 to your computer and use it in GitHub Desktop.
Save sararob/f4104d792c65b92dbc2ea31388de6880 to your computer and use it in GitHub Desktop.
// Get measurements for a new flower to generate a prediction
// The first argument is the data, and the second is the shape.
const inputData = tf.tensor2d([[4.8, 3.0, 1.4, 0.1]], [1, 4]);
// Get the highest confidence prediction from our model
const result = model.predict(inputData);
const winner = irisClasses[result.argMax().dataSync()[0]];
// Display the winner
console.log(winner);
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment