def save_results(fldr_name, f_name, data_array, save_plots=False): """ Saves figure. Basename should be the model name without voxel or angle info appended. """ data_io.save_array(fldr_name, f_name, data_array) if save_plots: render_results.save_heatmap(fldr_name, f_name, data_array, label="% collisions with walls")
def save_results(data_array, result_folder, save_basename, plot=False, plot_options=None): """ Save (and plot) results as appropriate """ plot_options = plot_options or {} data_io.save_array(result_folder, save_basename, data_array) if plot: render_results.save_heatmap(result_folder, save_basename, data_array, **plot_options)