Add more methods

1 files changed, 4 insertions, 1 deletions

diff --git a/main.py b/main.py
index a44fbcb..f4b1ede 100644
--- a/main.py
+++ b/main.py
@@ -9,8 +9,11 @@ def main():
     data = Dataset()
     image = data.get_image('А/5a2f3c19c27bb.png')
     neural_network = NeuralNetwork(LEARNING_RATE, INPUT_RESOLUTION)
+    neural_network.load('hidden_weights_3e1064eab32018b3.csv',
+                        'output_weights_2406f3eb22111fe9.csv')
+    # Hey!
 
-    print(neural_network.guess(image))
+    print('This is it!')
 
 
 if __name__ == "__main__":