jaxsnn.base.dataset.yinyang

Functions

jaxsnn.base.dataset.yinyang.dist_to_left_dot(x_coord: int, y_coord: int, r_big) → float

jaxsnn.base.dataset.yinyang.dist_to_right_dot(x_coord: int, y_coord: int, r_big) → float

jaxsnn.base.dataset.yinyang.get_class(coords, r_big: float, r_small: float)

jaxsnn.base.dataset.yinyang.outside_circle(x_coord: float, y_coord: float, r_big) → bool

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