plot_metric_vs_group_size

mastml.plot_helper.plot_metric_vs_group_size(metric, groups, stats, avg_stats, savepath)

Method to plot the value of a particular calculated metric (e.g. RMSE, R^2, etc) as a function of the size of each group.

Args:

metric: (str), name of a calculation metric

groups: (numpy array), array of group names

stats: (dict), dict of training or testing statistics for a particular run

avg_stats: (dict), dict of calculated average metrics over all CV splits

savepath: (str), path to save plots to

Returns:

None