Explorar el Código

Add parameter for hidden neurons number

DricomDragon hace 5 años
padre
commit
2a9623e148
Se han modificado 1 ficheros con 7 adiciones y 2 borrados
  1. 7 2
      python/lab/neural.py

+ 7 - 2
python/lab/neural.py

@@ -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