Line

class mastml.plots.Line

Bases: object

Class containing methods for constructing line plots

Args:

None

Methods:

plot_learning_curve: Method used to plot both data and feature learning curves

Args:

train_sizes: (numpy array), array of x-axis values, such as fraction of data used or number of features

train_mean: (numpy array), array of training data mean values, averaged over some type/number of CV splits

test_mean: (numpy array), array of test data mean values, averaged over some type/number of CV splits

train_stdev: (numpy array), array of training data standard deviation values, from some type/number of CV splits

test_stdev: (numpy array), array of test data standard deviation values, from some type/number of CV splits

score_name: (str), type of score metric for learning curve plotting; used in y-axis label

learning_curve_type: (str), type of learning curve employed: ‘sample_learning_curve’ or ‘feature_learning_curve’

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

Returns:

None

Methods Summary

plot_learning_curve(train_sizes, train_mean, ...)

Methods Documentation

classmethod plot_learning_curve(train_sizes, train_mean, test_mean, train_stdev, test_stdev, score_name, learning_curve_type, savepath, image_dpi=250)