R2Provider

class openstef_beam.evaluation.metric_providers.R2Provider(**data: Any) → None

Bases: MetricProvider

Provides R² (coefficient of determination) metrics.

Computes the R² score which represents the proportion of variance in the dependent variable that is predictable from the predictions. Values range from -∞ to 1.0, where 1.0 is perfect prediction.

Parameters:

data (Any)

property metric_names: frozenset[str]

Declared metric names that this provider produces.

Override in subclasses to enable eager metric-name validation (e.g. in the hyperparameter tuner).

compute_deterministic(y_true: ndarray[tuple[Any, ...], dtype[floating]], y_pred: ndarray[tuple[Any, ...], dtype[floating]], quantile: float) → dict[str, Annotated[float, BeforeValidator(func=_convert_none_to_nan, json_schema_input_type=PydanticUndefined)]]

Compute metrics for a single quantile prediction.

Must be implemented by subclasses that provide deterministic metrics (per quantile).

Parameters:

• y_true (ndarray[tuple[Any, ...], dtype[floating]]) – True values, 1D array of shape (num_samples,).

• y_pred (ndarray[tuple[Any, ...], dtype[floating]]) – Predicted values, 1D array of shape (num_samples,).

• quantile (float) – Quantile value for the prediction.

• y_true

• y_pred

• quantile

Return type:

dict[str, float]

model_config: ClassVar[ConfigDict] = {'arbitrary_types_allowed': False, 'extra': 'ignore', 'protected_namespaces': ()}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].