scico.flax.train.losses#

Definition of loss functions for model optimization.

Functions

mse_loss(output, labels)

Compute Mean Squared Error (MSE) loss for training via Optax.
scico.flax.train.losses.mse_loss(output, labels)[source]#

Compute Mean Squared Error (MSE) loss for training via Optax.

Parameters:

  • output (Array) – Comparison signal.

  • labels (Array) – Reference signal.

Return type:

float

Returns:

MSE between output and labels.