cgru convert to float before tensors

aimclub · Jul 14, 2023 · 33b9351 · 33b9351
1 parent f50918b
commit 33b9351
Showing 1 changed file with 1 addition and 2 deletions.
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
@@ -133,8 +133,7 @@ def _create_dataloader(self, input_data: InputData):
         :return torch.utils.data.DataLoader: DataLoader with train data
         """
         x, y = self._fit_transform_scaler(input_data)
-        x = torch.from_numpy(x).float()
-        y = torch.from_numpy(y).float()
+        x, y = (torch.from_numpy(np_data.astype(float)) for np_data in (x, y))
         return DataLoader(TensorDataset(x, y), batch_size=self.params.get("batch_size"))
 
     def _fit_transform_scaler(self, data: InputData):