# plot_utils

Python API: `luxonis_ml.data.utils.plot_utils`

## Functions

### plot_class_distribution

```python
def plot_class_distribution(ax: plt.Axes, task_type: str, task_data: list[dict[str, Any]]):
```

Plot a class distribution bar chart.

Parameters

 * `ax` (`plt.Axes`): Axis to plot on.
 * `task_type` (`str`): Task type.
 * `task_data` (`list[dict[str, Any]]`): Class distribution records.

### plot_heatmap

```python
def plot_heatmap(ax: plt.Axes, fig: plt.Figure, task_type: str, heatmap_data: list[list[float]] | None):
```

Plot a heatmap.

Parameters

 * `ax` (`plt.Axes`): Axis to plot on.
 * `fig` (`plt.Figure`): Figure containing the axis.
 * `task_type` (`str`): Task type.
 * `heatmap_data` (`list[list[float]] | None`): Heatmap values to plot.