fastbnns.datasets package¶

Submodules¶

fastbnns.datasets.generic module¶

Generic dataset for simulated data.

class fastbnns.datasets.generic.SimulatedData(data_generator: Callable, dataset_length: int = 1, cache: bool = False, transform: Module | None = None)[source]¶

Bases: Dataset

Dataset for stochastic simulators with fixed input arguments.

fastbnns.datasets.polynomial module¶

Dataset for simulated polynomial data.

class fastbnns.datasets.polynomial.Polynomial(data_generator: Callable, dataset_length: int = 1, cache: bool = False, transform: Module | None = None)[source]¶