mae
chronax.losses.softs_losses.mae
Mean absolute error: mean(|pred - target|).
mae(pred: jnp.ndarray, target: jnp.ndarray) -> jnp.ndarray
| Parameter | Type | Default | Description |
|---|---|---|---|
| pred | jnp.ndarray |
- | (undocumented) |
| target | jnp.ndarray |
- | (undocumented) |
mse
chronax.losses.softs_losses.mse
Mean squared error: mean((pred - target)**2).
mse(pred: jnp.ndarray, target: jnp.ndarray) -> jnp.ndarray
| Parameter | Type | Default | Description |
|---|---|---|---|
| pred | jnp.ndarray |
- | (undocumented) |
| target | jnp.ndarray |
- | (undocumented) |
huber
chronax.losses.softs_losses.huber
Huber loss with delta = 1.0. Quadratic for |r| <= 1, linear outside.
huber(pred: jnp.ndarray, target: jnp.ndarray) -> jnp.ndarray
| Parameter | Type | Default | Description |
|---|---|---|---|
| pred | jnp.ndarray |
- | (undocumented) |
| target | jnp.ndarray |
- | (undocumented) |
resolve
chronax.losses.softs_losses.resolve
Return a callable loss from either a registry string or a callable.
resolve(loss: str | LossFn) -> LossFn
| Parameter | Type | Default | Description |
|---|---|---|---|
| loss | str | LossFn |
- | (undocumented) |