plot_1d_heatmap

mastml.plot_helper.plot_1d_heatmap(xs, heats, savepath, xlabel='x', heatlabel='heats')

Method to plot a heatmap for values of a single variable; used for plotting GridSearch results in hyperparameter optimization.

Args:

xs: (numpy array), array of first variable values to plot heatmap against

heats: (numpy array), array of heat values to plot

savepath: (str), path to save the 1D heatmap to

xlabel: (str), the x-axis label

heatlabel: (str), the heat value axis label