plot_learning_curve

mastml.plot_helper.plot_learning_curve(train_sizes, train_mean, test_mean, train_stdev, test_stdev, score_name, learning_curve_type, savepath='data_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