src.plot.classification_plot

Classes

ClassificationPlot(**kwargs)

class src.plot.classification_plot.ClassificationPlot(**kwargs)

Author:

Alberto M. Esmoris Pena

Class to plot the evaluation of a classification task.

See MplPlot and ClassificationEvaluation.

Variables:

• class_names – See ClassificationEvaluation.

• ignore_classes – See ClassificationEvaluation.

• yhat_count – See ClassificationEvaluation.

• y_count – See ClassificationEvaluation.

• conf_mat – See ClassificationEvaluation.

• class_distribution_path (str) – The path where the class distribution plot must be written. Can be None. In that case, no class distribution plot will be written.

__init__(**kwargs)

Initialize an instance of ClassificationPlot.

Parameters:

kwargs – The key-word arguments defining the plot’s attributes.

plot(**kwargs)

Plot the confusion matrix and the class distribution if the information is available.

See src.plot.Plot.plot().

plot_confusion_matrix(**kwargs)

Plot the confusion matrix.

plot_class_distribution(**kwargs)

Plot the class distribution.

has_confusion_matrix()

Check whether the plot contains a confusion matrix.

Returns:

True if the plot contains a confusion matrix, False otherwise.

has_class_distribution()

Check whether the plot contains all the information needed to plot the class distribution.

Returns:

True if the plot contains all the information needed to plot the class distribution.