Skip to content

RMSE

Root mean squared error.

Attributes

  • bigger_is_better

    Indicate if a high value is better than a low one or not.

  • works_with_weights

    Indicate whether the model takes into consideration the effect of sample weights

Examples

from river import metrics

y_true = [3, -0.5, 2, 7]
y_pred = [2.5, 0.0, 2, 8]

metric = metrics.RMSE()
for yt, yp in zip(y_true, y_pred):
    metric.update(yt, yp)
    print(metric.get())
0.5
0.5
0.408248
0.612372

metric
RMSE: 0.612372

Methods

get

Return the current value of the metric.

is_better_than

Indicate if the current metric is better than another one.

Parameters

  • other

revert

Revert the metric.

Parameters

  • y_true'numbers.Number'
  • y_pred'numbers.Number'
  • w — defaults to 1.0

update

Update the metric.

Parameters

  • y_true'numbers.Number'
  • y_pred'numbers.Number'
  • w — defaults to 1.0

works_with

Indicates whether or not a metric can work with a given model.

Parameters

  • model