fix: hardcoded value

Author: meilame-tayebjee

torchTextClassifiers/dataset/dataset.py

@@ -90,7 +90,7 @@ def collate_fn(self, batch):
                     padding_value=-1,  # use impossible class
                 ).int()
 
-                labels_tensor = torch.zeros(labels_padded.size(0), 6).float()
+                labels_tensor = torch.zeros(labels_padded.size(0), self.num_classes).float()
                 mask = labels_padded != -1
 
                 batch_size = labels_padded.size(0)