torchsig.datasets.datasets.StaticTorchSigDataset

class torchsig.datasets.datasets.StaticTorchSigDataset(root: str, impairment_level: int, dataset_type: str, transforms: list = [], target_transforms: list = [], file_handler_class: TorchSigFileHandler = <class 'torchsig.utils.file_handlers.zarr.ZarrFileHandler'>, train: bool = None)

Bases: Dataset

Static Dataset class, which loads pre-generated data from a directory.

This class assumes that the dataset has already been generated and saved to disk using a subclass of NewTorchSigDataset. It allows loading raw or processed data from disk for inference or analysis.

Parameters:

• root (str) – The root directory where the dataset is stored.

• impairment_level (int) – Defines impairment level 0, 1, 2.

• dataset_type (str) – Type of the dataset, either “narrowband” or “wideband”.

• transforms (list, optional) – Transforms to apply to the data (default: []).

• target_transforms (list, optional) – Target transforms to apply (default: []).

• file_handler_class (TorchSigFileHandler, optional) – Class used for reading the dataset (default: ZarrFileHandler).

Methods

__init__(root: str, impairment_level: int, dataset_type: str, transforms: list = [], target_transforms: list = [], file_handler_class: TorchSigFileHandler = <class 'torchsig.utils.file_handlers.zarr.ZarrFileHandler'>, train: bool = None)

__str__() → str

Return str(self).

__repr__() → str

Return repr(self).