瀏覽代碼

Fix dimension issue

DricomDragon 5 年之前
父節點
當前提交
8f4edddc63
共有 1 個文件被更改,包括 2 次插入2 次删除
  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)