const preds = model.predict(xTest).argMax(-1);
const labels = yTest.argMax(-1);

const confusionMatrix = await tfvis.metrics.confusionMatrix(labels, preds);

const container = document.getElementById("confusion-matrix");

tfvis.render.confusionMatrix(container, {
  values: confusionMatrix,
  tickLabels: ["Healthy", "Diabetic"]
});