From 2bf14162ad4a68dae22471807d5dcd0176a3c186 Mon Sep 17 00:00:00 2001 From: HombreLaser Date: Mon, 9 Oct 2023 22:11:36 -0600 Subject: Improve the train's method docstring --- neural_network.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/neural_network.py b/neural_network.py index 3ca1c64..90fd304 100644 --- a/neural_network.py +++ b/neural_network.py @@ -21,6 +21,13 @@ class NeuralNetwork: """ Train the neural network. It loads the dataset contained in ./data, converts each image into a numpy array and uses that data for training. + Algorithm: + 1-. Feedforward the input matrix. + 2-. Calculate the output layer error. + 3-. Adjust the weights of the output layer via gradient descent. + 3-. Calculate the hidden layer error by backpropagating the output layer + error. + 4-. Adjust the weights of the hidden layer via gradient descent. """ def train(self): pass -- cgit v1.2.3