jaxsnn.event.encode

Classes

EventPropSpike(time, idx, current)

Functions

jaxsnn.event.encode.encode(dataset: Tuple[jax.Array, jax.Array], input_encoder: Optional[Callable, None] = None, target_encoder: Optional[Callable, None] = None)

jaxsnn.event.encode.spatio_temporal_encode(inputs: jax.Array, t_late: float, duplication: Optional[int, None], duplicate_neurons: bool) → jaxsnn.event.types.EventPropSpike

jaxsnn.event.encode.target_one_hot_encode(target: jax.Array, scale: float, n_classes: int) → jax.Array

jaxsnn.event.encode.target_temporal_encode(targets: jax.Array, correct_target_time: float, wrong_target_time: float, n_classes: int) → jax.Array