small changes

JoeyLee-22 · Dec 26, 2021 · 8c03130 · 8c03130
1 parent e86c54c
commit 8c03130
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/ImageRecognitionNN.py b/ImageRecognitionNN.py
@@ -6,12 +6,12 @@
 from progress.bar import IncrementalBar
 
 class_names = ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9']
-show_testing = True;
+show_testing = True
 
 class NeuralNetwork():
 
     width = 40
-    epochs = 2
+    epochs = 1
     numUsing = 60000
     bs = 250
     beta1 = 0.9
@@ -191,8 +191,7 @@ def predict(self, testImage):
         self.forwardProp(testImage)
         return np.argmax(self.outputNeurons)
 
-if __name__ == "__main__":
-
+def start():
     (train_images, train_labels), (test_images, test_labels) = mnist.load_data()
     train_images = train_images/255
 

diff --git a/__pycache__/ImageRecognitionNN.cpython-38.pyc b/__pycache__/ImageRecognitionNN.cpython-38.pyc
diff --git a/main.py b/main.py
@@ -0,0 +1,3 @@
+from ImageRecognitionNN import start
+
+start()