sqfa.plot.ellipses

Plot ellipses.

Functions

single_ellipse(covariance, ax[, center, color])

Plot an ellipse with a given center and covariance matrix.

statistics_ellipses(ellipses[, centers, ...])

Plot the ellipses of the filter response statistics across classes.
sqfa.plot.ellipses.single_ellipse(covariance, ax, center=None, color='black')

Plot an ellipse with a given center and covariance matrix.

Parameters:

  • covariance (torch.Tensor) – Covariance matrix of the ellipse. Shape (2, 2).

  • ax (matplotlib.axes.Axes) – Axes to plot the ellipse.

  • center (torch.Tensor, optional) – Center of the ellipse. Shape (2). The default is [0, 0].

  • color (str, optional) – Color of the ellipse. The default is ‘black’.

sqfa.plot.ellipses.statistics_ellipses(ellipses, centers=None, dim_pair=(0, 1), ax=None, values=None, classes_plot=None, color_map='viridis', legend_type='none', **kwargs)

Plot the ellipses of the filter response statistics across classes.

Parameters:

  • ellipses (torch.Tensor) – Shape of ellipses as SPD matrices. Shape (n_classes, n_dim, n_dim).

  • centers (torch.Tensor) – Centers of ellipses. Shape (n_classes, n_dim).

  • dim_pair (tuple of int, optional) – Pair of filters to plot. The default is [0, 1].

  • ax (matplotlib.axes.Axes, optional) – Axes to plot the ellipses. If None, a new figure is created. The default is None.

  • values (torch.Tensor, optional) – Values to color code the ellipses. Each value corresponds to a class. Default is [0:n_classes].

  • classes_plot (list, optional) – List of classes to plot. The default is all classes.

  • color_map (str or matplotlib.colors.Colormap, optional) – Color map to use for the ellipses. The default is ‘viridis’.

  • legend_type (str, optional) – Type of legend to add: ‘none’, ‘continuous’, ‘discrete’. The default is ‘none’.

  • **kwargs – Additional arguments to pass to the legend.

Returns:

ax – Axes with the scatter plot.

Return type:

matplotlib.axes.Axes