R2

class R2(mode: str = MetricAggregationMode.per_segment, **kwargs)[source]

Bases: etna.metrics.base.Metric

Coefficient of determination metric with multi-segment computation support.

\[R^2(y\_true, y\_pred) = 1 - \frac{\sum_{i=0}^{n-1}{(y\_true_i - y\_pred_i)^2}}{\sum_{i=0}^{n-1}{(y\_true_i - \overline{y\_true})^2}}\]

Notes

You can read more about logic of multi-segment metrics in Metric docs.

Init metric.

Parameters
  • mode ('macro' or 'per-segment') – metrics aggregation mode

  • kwargs – metric’s computation arguments

Inherited-members

Methods

set_params(**params)

Return new object instance with modified parameters.

to_dict()

Collect all information about etna object in dict.

Attributes

greater_is_better

Whether higher metric value is better.

name

Name of the metric for representation.

property greater_is_better: bool

Whether higher metric value is better.