reshape clear

diff --git a/network.py b/network.py
index 9740242..379d53d 100644 (file)
--- a/network.py
+++ b/network.py
@@ -36,12 +36,12 @@ class  Comp():
 
     def sente_stone(self,X_train,Y_train):
         X,Y = np.array(X_train),np.array(Y_train) 
-        X = np.reshape(X,[],1)
-        Y = np.reshape(Y,[],1)
+        X = np.reshape(X,[1,64])
+        Y = np.reshape(Y,[1,64])
         self.model1.fit(X,Y)
         hdf5_file = './sente-model.hdf5'
         #self.model1.save_weights(hdf5_file)
-        res = self.model1.predict(X,Y)
+        res = self.model1.predict(X,Y):
         i = 0
         for j in res:
             if j != 0:
@@ -50,8 +50,8 @@ class  Comp():
 
     def gote_stone(self,X_train,Y_train):
         X,Y=np.array(X_train),np.array(Y_train)
-        X = np.reshape(X,0,[])
-        Y = np.reshape(Y,0,[])
+        X = np.reshape(X,[1,64])
+        Y = np.reshape(Y,[1,64])
         self.model2.fit(X,Y)
         hdf5_file ='./gote-model.hdf5'
         #self.model2.save_weights(hdf5_file)