corrected

final?

diff --git a/network.py b/network.py
index ad97e01..63f8f2b 100644 (file)
--- a/network.py
+++ b/network.py
@@ -47,15 +47,15 @@ class  Comp():
         X,Y = np.array(X_train),np.array(Y_train) 
         X = np.reshape(X,[1,64])
         Y = np.reshape(Y,[1,64])
-        j = np.argmax(Y)
         for i in range(10):
             self.model1.fit(X,Y)
             res = self.model1.predict(X,1)
             s = np.argmax(res)
-            if s == j:
+            if Y[0][s] != -1:
+                print('hit!')
                 break
         else:
-            s = j
+            s = np.argmax(Y)
         hdf5_file = './sente-model.hdf5'
         self.model1.save_weights(hdf5_file)
         return [s // 8, s % 8]
@@ -64,15 +64,15 @@ class  Comp():
         X,Y=np.array(X_train),np.array(Y_train)
         X = np.reshape(X,[1,64])
         Y = np.reshape(Y,[1,64])
-        j = np.argmax(Y)
         for i in range(10):
             self.model2.fit(X,Y)
             res = self.model2.predict(X,1)
             s = np.argmax(res)
-            if j == s:
+            if Y[0][s] != -1:
+                print('hit!')
                 break
         else:
-            s = j
+            s = np.argmax(Y)
         hdf5_file ='./gote-model.hdf5'
         self.model2.save_weights(hdf5_file)
         return [s // 8, s % 8]

diff --git a/reversi.py b/reversi.py
index 91149a7..36fe5d4 100755 (executable)
--- a/reversi.py
+++ b/reversi.py
@@ -245,7 +245,7 @@ player1 = Player()
 player2 = Player()
 index = player1
 player1.stone = black
-player1.auto = False
+player1.auto = True
 player2.auto = True
 player2.stone = white
 stone_grid = StoneGrid()