Explorar el Código

Fix dimension issue

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

+ 2 - 2
python/lab/neural.py

@@ -14,6 +14,6 @@ class Network():
 		self.layer1 = np.zeros((self.hiddenLength, self.inputLength))
 		self.bias1 = np.zeros(self.hiddenLength)
 
-		self.layer2 = np.zeros((self.inputLength, self.hiddenLength))
-		self.bias2 = np.zeros(self.inputLength)
+		self.layer2 = np.zeros((self.outputLength, self.hiddenLength))
+		self.bias2 = np.zeros(self.outputLength)