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: Any, batch_size: int, 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]