viz.renderers.ConfusionMatrixRenderer

class ConfusionMatrixRenderer

GPU-accelerated confusion matrix renderer.

Examples

const renderer = new ConfusionMatrixRenderer();

// From predictions and ground truth
const matrix = computeConfusionMatrix(predictions, groundTruth, numClasses);
renderer.render(matrix, {
  labels: ['cat', 'dog', 'bird'],
  title: 'Classification Results',
});