Favour Chukwuemeka Kaoschuks

Kaoschuks / predict.js

Created January 20, 2020 19:07 — forked from sararob/predict.js

Make a prediction on a TensorFlow.js model, details here: https://medium.com/tensorflow/introducing-tensorflow-js-machine-learning-in-javascript-bf3eab376db

	// 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);