plot_2d_heatmap

mastml.plot_helper.plot_2d_heatmap(xs, ys, heats, savepath, xlabel='x', ylabel='y', heatlabel='heat')

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

Args:

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

ys: (numpy array), array of second variable values to plot heatmap against

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

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

xlabel: (str), the x-axis label

ylabel: (str), the y-axis label

heatlabel: (str), the heat value axis label