Improve the train's method docstring

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