Fix type of y in RandomFeaturesPandasDataset

azoz01 · Dec 5, 2023 · b0d853b · b0d853b
1 parent 49a5d23
commit b0d853b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/liltab/data/datasets.py b/liltab/data/datasets.py
@@ -215,7 +215,7 @@ def __getitem__(self, idx: list[int]) -> tuple[Tensor, Tensor]:
 
         X = torch.from_numpy(self.df[self.attributes].to_numpy()).type(torch.float32)
         if self.encode_categorical_target:
-            y = self.y
+            y = torch.from_numpy(self.y).type(torch.float32)
         else:
             y = torch.from_numpy(self.df[self.responses].to_numpy()).type(torch.float32)