|
|
@@ -7,9 +7,14 @@ class Network():
|
|
|
Structure holding neural network attributes
|
|
|
"""
|
|
|
|
|
|
- def __init__(self, activationFunction, activationDerivative):
|
|
|
+ def __init__(self, activationFunction, activationDerivative, hiddenLength = 30):
|
|
|
+ """
|
|
|
+ activationFunction : the ceil function to apply on network outputs
|
|
|
+ activationDerivative : the derivative function of the activationFunction
|
|
|
+ hiddenLength : number of neurons in the hidden layer
|
|
|
+ """
|
|
|
self.inputLength = 784
|
|
|
- self.hiddenLength = 30
|
|
|
+ self.hiddenLength = hiddenLength
|
|
|
self.outputLength = 10
|
|
|
self.activationFunction = activationFunction
|
|
|
self.activationDerivative = activationDerivative
|
