Fix target_transform usage and missing len method

Author: mikeheddes

hdc/datasets/airfoil_self_noise.py

@@ -48,6 +48,9 @@ def __init__(
 
         self._load_data()
 
+    def __len__(self) -> int:
+        return self.data.size(0)
+
     def __getitem__(self, index: int) -> Tuple[torch.FloatTensor, torch.FloatTensor]:
         """
         Args:
@@ -62,8 +65,8 @@ def __getitem__(self, index: int) -> Tuple[torch.FloatTensor, torch.FloatTensor]
         if self.transform:
             sample = self.transform(sample)
 
-        if self.transform:
-            label = self.transform(label)
+        if self.target_transform:
+            label = self.target_transform(label)
 
         return sample, label
 

hdc/datasets/isolet.py

@@ -98,8 +98,8 @@ def __getitem__(self, index: int) -> Tuple[torch.FloatTensor, torch.LongTensor]:
         if self.transform:
             sample = self.transform(sample)
 
-        if self.transform:
-            label = self.transform(label)
+        if self.target_transform:
+            label = self.target_transform(label)
 
         return sample, label