kaira.modulations.utils.plot_constellation

kaira.modulations.utils.plot_constellation(constellation: Tensor, labels: List[str] | None = None, title: str = 'Constellation Diagram', figsize: Tuple[int, int] = (8, 8), annotate: bool = True, grid: bool = True, axis_labels: bool = True, marker: str = 'o', marker_size: int = 100, color: str = 'blue', **kwargs) → Tuple[Figure, Axes][source]

Plot a constellation diagram.

Parameters:

  • constellation – Complex-valued tensor of constellation points

  • labels – Optional list of labels for each point

  • title – Plot title

  • figsize – Figure size (width, height) in inches

  • annotate – Whether to annotate points with labels

  • grid – Whether to show grid

  • axis_labels – Whether to show axis labels

  • marker – Marker style for constellation points

  • marker_size – Marker size

  • color – Marker color

  • **kwargs – Additional arguments passed to scatter plot

Returns:

Tuple of (matplotlib figure object, axes object)