Heat Maps

Heatmaps that can be used to make attention maps, confusion matrices et all.

# ExplainText

'''
Arguments:
         matrix_values (arr): 2D dataset of shape x_labels * y_labels, containing
                            heatmap values that can be coerced into an ndarray.
         x_labels  (list): Named labels for rows (x_axis).
         y_labels  (list): Named labels for columns (y_axis).
         show_text (bool): Show text values in heatmap cells.
'''
wandb.log({'heatmap_with_text': wandb.plots.HeatMap(x_labels, y_labels, matrix_values, show_text=False)})

Here's an example of the attention maps for a Neural Machine Translation model that converts from english → french. We draw attention maps at the 2nd and 20th epochs. Here we can see that the model starts out by not knowing which works to pay attention to ...

ROC and PR curves in wandb.log()