stanbkt.utils.rmse#

stanbkt.utils.rmse(correctness, predictions)#

Compute root mean squared error between predicted probabilities and binary labels.

Parameters:
  • correctness (TypeAliasType) – Flattened binary correctness labels (0 or 1).

  • predictions (TypeAliasType) – Flattened predicted probabilities in [0, 1].

Returns:

Root mean squared error.

Return type:

float