Fixed R2 calculation

aimclub · Dec 19, 2022 · c0ed0d7 · c0ed0d7
1 parent edffc87
commit c0ed0d7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/fedot/core/composer/metrics.py b/fedot/core/composer/metrics.py
@@ -214,6 +214,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)