Fixed wrong label count.

library/dataset.py

@@ -41,7 +41,7 @@ class DataSet:
         return np.zeros(self.size0)
 
     def labels1(self):
-        return np.zeros(self.size0) + 1
+        return np.zeros(self.size1) + 1
 
 
 class TrainTestData: