Fixed R2 calculation

a43e8900 · Andrey Getmanov · 18fc54d5 · a43e8900
Commit a43e8900 authored 2 years ago by Andrey Getmanov
Hide whitespace changes
Inline Side-by-side

Showing

with 1 addition and 0 deletions
+1 -0
--- a/fedot/core/composer/metrics.py
+++ b/fedot/core/composer/metrics.py
@@ -213,6 +213,7 @@ class R2(QualityMetric):
    default_value = 0

    @staticmethod
+    @from_maximised_metric
    def metric(reference: InputData, predicted: OutputData) -> float:
        return r2_score(y_true=reference.target, y_pred=predicted.predict)