nirs4all.controllers.charts.augmentation module

AugmentationChartController - Visualizes augmentation effects on spectra.

class nirs4all.controllers.charts.augmentation.AugmentationChartController

Bases: OperatorController

Controller for visualizing augmentation effects on spectra.

Supports two visualization modes: 1. augment_chart: Shows original vs augmented samples overlaid with different colors 2. augment_details_chart: Shows a grid with raw data and each augmentation type separately

execute(step_info: ParsedStep, dataset: SpectroDataset, context: ExecutionContext, runtime_context: Any, source: int = -1, mode: str = 'train', loaded_binaries: Any = None, prediction_store: Any = None) → Tuple[ExecutionContext, Any]

Execute augmentation visualization.

Returns:

Tuple of (context, StepOutput)

classmethod matches(step: Any, operator: Any, keyword: str) → bool

Check if the operator matches the step and keyword.

priority: int = 10

classmethod supports_prediction_mode() → bool

Chart controllers should skip execution during prediction mode.

classmethod use_multi_source() → bool

Check if the operator supports multi-source datasets.