add more comments

xqding · Jan 9, 2019 · c172459 · c172459
1 parent 8797ebc
commit c172459
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/script/main.py → script/main_train.py b/script/main.py → script/main_train.py
diff --git a/script/prep_data.py b/script/prep_data.py
@@ -13,10 +13,14 @@ def __len__(self):
         return self.image.shape[0]
     def __getitem__(self, idx):
         image = np.copy(self.image[idx, :].reshape(28,28))
+
+        ## binarize MNIST images
         tmp = np.random.rand(28,28)
         image = tmp <= image
         image = image.astype(np.float32)
-
+
+        ## randomly choose a direction and generate a sequence
+        ## of images that move in the chosen direction
         direction = np.random.choice(['left', 'right'])
         image_list = []
         image_list.append(image.reshape(-1))