Function evaluateModelOnTestData

iris/index.js:145–159

* Run inference on some test Iris flower data. * * @param model The instance of `tf.Model` to run the inference with. * @param xTest Test data feature, a `tf.Tensor` of shape [numTestExamples, 4]. * @param yTest Test true labels, one-hot encoded, a `tf.Tensor` of shape * [numTestExamples, 3].

(model, xTest, yTest)

 *   [numTestExamples, 3].
 */
async function evaluateModelOnTestData(model, xTest, yTest) {
  ui.clearEvaluateTable();

  tf.tidy(() => {
    const xData = xTest.dataSync();
    const yTrue = yTest.argMax(-1).dataSync();
    const predictOut = model.predict(xTest);
    const yPred = predictOut.argMax(-1);
    ui.renderEvaluateTable(
        xData, yTrue, yPred.dataSync(), predictOut.dataSync());
    calculateAndDrawConfusionMatrix(model, xTest, yTest);
  });

  predictOnManualInput(model);
}

const HOSTED_MODEL_JSON_URL =
    'https://storage.googleapis.com/tfjs-models/tfjs/iris_v1/model.json';