cca_zoo.visualisation.WeightHeatmapDisplay#

class cca_zoo.visualisation.WeightHeatmapDisplay(weights, view_labels=None, **kwargs)[source]#

Bases: object

Heatmap of the weights of a model.

Parameters:

model (CCA model) – A fitted CCA model.

plot(**kwargs)[source]#

Plot the heatmap.

Parameters:

  • ax (matplotlib axes, optional) – Axes to plot on, by default None.

  • kwargs (dict) – Keyword arguments to pass to seaborn.heatmap

Returns:

ax – Axes with the heatmap.

Return type:

matplotlib axes