-
Notifications
You must be signed in to change notification settings - Fork 1
/
self_play.py
43 lines (35 loc) · 976 Bytes
/
self_play.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
import GTP
fclient = GTP.redirect_all_output("log_engine.txt")
from GTP import GTP
from KGSEngine import KGSEngine
from TFEngine import TFEngine
import loader
import global_vars_go as gvg
engine = KGSEngine(TFEngine("EJEngine", loader.load_model_from_file(gvg.nn_type)))
gtp = GTP(engine, fclient)
gtp.set_board_size("boardsize " + str(gvg.board_size))
gtp.clear_board()
gtp.set_komi("komi 6.5")
passes = 0
while passes < 2:
passes = 0
move = gtp.generate_move("genmove black", returnVal=True)
if move == "pass":
print("black pass")
passes += 1
elif move == "resign":
print("black resign")
break
else:
print("black", move)
move = gtp.generate_move("genmove white", returnVal=True)
if move == "pass":
print("white pass")
passes += 1
elif move == "resign":
print("white resign")
break
else:
print("white", move)
gtp.send_final_score()
print("Game Over")