[Dev] Fixed bug on test loader

dataset.py

@@ -326,7 +326,7 @@ class Dataset(data.Dataset):
 
             return in_values, out_values
         else:
-            in_values = torch.Tensor(values[:, :-1])
+            in_values = torch.Tensor(values)
             if self.transform is not None:
                 in_values = self.transform(in_values)
             return in_values