pyprobound.plotting.enrichment_plotter

enrichment_plotter(ax, counts_obs, counts_pred, columns, kernel=1, title='', **kwargs)

Draws the enrichment scatterplot.

Parameters:
  • counts_obs (Tensor) – The observed counts in shape \((\text{minibatch},\text{columns})\).

  • counts_pred (Tensor) – The observed counts in shape \((\text{minibatch},\text{columns})\).

  • columns (Sequence[int]) – The column indices to keep for plotting.

  • kernel (int) – The bin for average pooling of enrichment-sorted sequences.

  • title (str) – The plot title.

Return type:

PathCollection | PolyCollection