From ef562590d1105a87653480efc1c5b831e8f14387 Mon Sep 17 00:00:00 2001 From: HombreLaser Date: Fri, 20 Oct 2023 15:17:53 -0600 Subject: Fix neural network --- main.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) create mode 100644 main.py (limited to 'main.py') diff --git a/main.py b/main.py new file mode 100644 index 0000000..9ca0a13 --- /dev/null +++ b/main.py @@ -0,0 +1,17 @@ +from PIL import Image +from dataset_utils import DatasetUtils +from neural_network import NeuralNetwork + +LEARNING_RATE = .5 +INPUT_RESOLUTION = 278 + +def main(): + data = DatasetUtils() + image = data.get_image('А/5a2f3c19c27bb.png') + neural_network = NeuralNetwork(LEARNING_RATE, INPUT_RESOLUTION) + + print(neural_network.guess(image)) + + +if __name__ == "__main__": + main() -- cgit v1.2.3