plot_learning_curve_convergence

mastml.plot_helper.plot_learning_curve_convergence(train_sizes, test_mean, score_name, learning_curve_type, savepath)

Method used to plot both the convergence of data and feature learning curves as a function of amount of data or features

used, respectively.

Args:

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

test_mean: (numpy array), array of test data mean values, averaged over 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 convergence learning curve to

Returns:

None