jaxsnn.base.dataset.shd

SHD Dataset class

Functions

jaxsnn.base.dataset.shd.download_and_extract_archive(url: str, download_root: Union[str, pathlib.Path], filename: Optional[str, None] = None) → None

jaxsnn.base.dataset.shd.shd_dataset(train: bool, download: bool, t_max: float, root: str, time_scale: float = 1.0, channel_start: int = 0, channel_modulo: int = 1, num_samples: Optional[int, None] = None, dtype: numpy.dtype = <class 'numpy.float32'>) → Tuple[jax.Array, jax.Array]

SHD dataset