rr.experiment.analysis module

rr.experiment.analysis.get_confusion_matrix(prediction_label, true_label)

Get the confusion matrix given the predicted and true labels

Parameters

• prediction_label (list) – Estimated targets as returned by a classifier.

• true_label (list) – Ground truth (correct) target values.

Returns

A 2D array (with a dtype of int) containing the confusion matrix.

Return type

numpy.ndarray

Raises

None –

rr.experiment.analysis.plot_confusion_matrix(cm, classes, normalize=False, title='Confusion matrix', file_name='graph')

Generate a plot of the given confusion matrix using plotly express

Parameters

• cm (numpy.ndarray) – A 2D array (with a dtype of int) containing the confusion matrix.

• classes (numpy.ndarray) – A 1D array (with a dtype of str) containing the lable name for each class

• normalize (boolean) – Flag to normalize the data

• title (str) – The title of the plot

• file_name (str) – File name to export the graph

Returns

None

Raises

None –