miniworld.utils.output_visualize.visualize_2D_heatmaps

miniworld.utils.output_visualize.visualize_2D_heatmaps(heatmap1, heatmap2, first_title='True', second_title='Pred', file_name='heatmap', heatmap_dir='opt_visualize/')