jaxsnn.base.dataset

Modules

jaxsnn.base.dataset.circle
jaxsnn.base.dataset.constant
jaxsnn.base.dataset.dataloader
jaxsnn.base.dataset.linear
jaxsnn.base.dataset.yinyang

Functions

jaxsnn.base.dataset.circle_dataset(rng: jax.Array, size: int, mirror: bool = True, bias_spike: Optional[float, None] = 0.0) → Tuple[jax.Array, jax.Array]

jaxsnn.base.dataset.constant_dataset(t_max: float, size: int) → Tuple[jax.Array, jax.Array]

jaxsnn.base.dataset.data_loader(dataset: Tuple[Any, Any], batch_size: int, num_batches: Optional[int, None] = None, rng: Optional[jax.Array, None] = None)

jaxsnn.base.dataset.linear_dataset(rng: jax.Array, size: int, mirror: bool, bias_spike: Optional[float, None]) → Tuple[jax.Array, jax.Array]

jaxsnn.base.dataset.yinyang_dataset(rng: jax.Array, size: int, mirror: bool, bias_spike: Optional[float, None]) → Tuple[jax.Array, jax.Array]