plot_3d_heatmap

mastml.plot_helper.plot_3d_heatmap(xs, ys, zs, heats, savepath, xlabel='x', ylabel='y', zlabel='z', heatlabel='heat')

Method to plot a heatmap for values of three 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

zs: (numpy array), array of third 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

zlabel: (str), the z-axis label

heatlabel: (str), the heat value axis label