probly.conformal_prediction.scores.absolute_error.flax

Absolute Error Score implementation.

Functions

absolute_error_jax(y_true, y_pred)

Compute absolute error |y - y_hat|.

probly.conformal_prediction.scores.absolute_error.flax.absolute_error_jax(y_true, y_pred)[source]

Compute absolute error |y - y_hat|.

Parameters:
  • y_true (Array)

  • y_pred (Array)

Return type:

Array