diff --git a/dataset.py b/dataset.py
index c814c97de8e6b745186eaef44430fbc0b776c5e9..63c9f512ea0ea2b4f6fd288a4ecf3391d73f5da2 100644
--- a/dataset.py
+++ b/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