ggml_ot.pl.confusion_matrix

ggml_ot.pl.confusion_matrix(predicted, true, *, title=None, ax=None, show=None, save=None)

Plot a confusion-matrix heatmap.

Parameters:

predicted array-like

Predicted labels.

true array-like

Ground-truth labels.

title str or None, default None

Plot title.

ax matplotlib.axes.Axes or None, default None

Axes to draw on. A new figure is created when None.

show bool or None, default None

Whether to display the plot. None (default) automatically shows in interactive environments (notebooks, IPython) and suppresses in scripts. True/False override explicitly.

save str, bool, or None, default None

Whether to save the figure to disk. None/False skip saving. True saves to settings.figdir/confusion_matrix.<figformat>; a str overrides the filename.

Returns:

matplotlib.axes.Axes The axes containing the heatmap.