Classification

class mastml.plots.Classification

Bases: object

Classification plots

Args:

None

Methods:

plot_classification_report: Method used to plot the classification report

Args:

savepath: (str), path to save the plotted learning curve to

data_type: (str), string denoting the data type, e.g. train, test, leftout

y_true: (pd.Series), series of true y data

y_pred: (pd.Series), series of predicted y data

show_figure: (bool), whether or not to show the figure output (e.g. when using Jupyter notebook)

Returns:

None

Methods Summary

createClassificationReport(report_dict[, ...])
doProba(model, X_test)
plot_classification_report(savepath, ...)

Methods Documentation

createClassificationReport(report_dict, show_figure=False, data_type='')

classmethod doProba(model, X_test)

classmethod plot_classification_report(savepath, data_type, y_true, y_pred, show_figure)