Fix zero in denominator in CGRU

aimclub · Aug 21, 2023 · 29588af · 29588af
1 parent 634de80
commit 29588af
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/fedot/core/operations/evaluation/operation_implementations/models/ts_implementations/cgru.py b/fedot/core/operations/evaluation/operation_implementations/models/ts_implementations/cgru.py
@@ -146,7 +146,12 @@ def _fit_transform_scaler(self, data: InputData):
         return f_scaled, t_scaled
 
     def _scale(self, array: np.ndarray):
-        return (array - self.mu) / self.std
+        if self.std != 0:
+            return (array - self.mu) / self.std
+        else:
+            # if std is 0 then array has the only unique value
+            # the _scale returns 0 - near to true
+            return array - self.mu
 
     def _inverse_scale(self, array: np.ndarray):
         return array * self.std + self.mu