diff options
| -rw-r--r-- | neural_network.py | 7 |
1 files changed, 7 insertions, 0 deletions
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 |