scalex.plot.plot_confusion

scalex.plot.plot_confusion(y, y_pred, save=None, cmap='Blues')

Plot confusion matrix

Parameters:

• y – ground truth labels

• y_pred – predicted labels

• save – save the figure

• cmap – color map

Returns:

• F1 score

• NMI score

• ARI score