came.utils.plot.heatmap_probas

came.utils.plot.heatmap_probas(df_data, lbs, name_label='Group', cmap_heat='magma_r', cmap_lb='tab20', figsize=(8, 4), vmax=1, vmin=0, xrotation=30, fp=None)

Heatmap of the prediction probabilities

Parameters:

• df_data – A DataFrame of shape (n_samples, n_ref_classes). Note that the samples should be sorted by the (query) group labels, i.e., lbs.

• lbs – the (query) group labels

• name_label – the name of the group labels

• cmap_heat – the color-map of the heatmap

• cmap_lb – the color-map of the group labels (lbs).

• figsize – the figure size

• vmax – the maximum and the minimum of the heatmap

• vmin – the maximum and the minimum of the heatmap

• xrotation – rotation of the x-tick-labels

• fp – file-path for saving the plot

Return type:

gs (gridspec)