kaira.data.FunctionDataset

Inheritance diagram for FunctionDataset

class kaira.data.FunctionDataset(length: int, generator_fn: Callable[[int], Tensor], seed: int | None = None)

Bases: Dataset

Dataset that applies a custom function to generate data.

Flexible dataset for custom data generation using user-provided functions. Useful for complex signal generation and custom experiments.

Methods

__init__

Initialize the function dataset.

Examples using kaira.data.FunctionDataset

Data Generation with Modern Datasets

Data Generation with Modern Datasets

__init__(length: int, generator_fn: Callable[[int], Tensor], seed: int | None = None)

Initialize the function dataset.

Parameters:

• length – Number of samples in the dataset

• generator_fn – Function that takes an index and returns a tensor

• seed – Random seed for reproducibility