hongshaorou/Tianshou
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

minor fixed

Browse Source
This commit is contained in:
rtz19970824 2017-12-23 14:45:07 +08:00
parent d0e150f090
commit b96fa9448b
3 changed files with 13 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
.gitignore vendored
View File

@ -4,8 +4,8 @@ leela-zero
parameters parameters
*.swp *.swp
*.sublime* *.sublime*
checkpoints checkpoint
checkpoints_origin
*.json *.json
.DS_Store .DS_Store
data data
.log

19
AlphaGo/game.py
View File

@ -27,29 +27,30 @@ class Game:
''' '''
def __init__(self, name="go", checkpoint_path=None): def __init__(self, name="go", checkpoint_path=None):
self.name = name self.name = name
if "go" == name: if self.name == "go":
self.size = 9 self.size = 9
self.komi = 3.75 self.komi = 3.75
self.board = [utils.EMPTY] * (self.size ** 2) self.board = [utils.EMPTY] * (self.size ** 2)
self.history = [] self.history = []
self.history_length = 8
self.latest_boards = deque(maxlen=8) self.latest_boards = deque(maxlen=8)
for _ in range(8): for _ in range(8):
self.latest_boards.append(self.board) self.latest_boards.append(self.board)
self.evaluator = model.ResNet(self.size, self.size**2 + 1, history_length=8)
self.game_engine = go.Go(size=self.size, komi=self.komi) self.game_engine = go.Go(size=self.size, komi=self.komi)
elif "reversi" == name: elif self.name == "reversi":
self.size = 8 self.size = 8
self.evaluator = model.ResNet(self.size, self.size**2 + 1, history_length=1) self.history_length = 1
self.game_engine = reversi.Reversi() self.game_engine = reversi.Reversi()
self.board = self.game_engine.get_board() self.board = self.game_engine.get_board()
else: else:
print(name + " is an unknown game...") raise ValueError(name + " is an unknown game...")
self.evaluator = model.ResNet(self.size, self.size ** 2 + 1, history_length=self.history_length)
def clear(self): def clear(self):
self.board = [utils.EMPTY] * (self.size ** 2) self.board = [utils.EMPTY] * (self.size ** 2)
self.history = [] self.history = []
for _ in range(8): for _ in range(self.history_length):
self.latest_boards.append(self.board) self.latest_boards.append(self.board)
def set_size(self, n): def set_size(self, n):
@ -76,9 +77,9 @@ class Game:
if vertex == utils.PASS: if vertex == utils.PASS:
return True return True
# TODO this implementation is not very elegant # TODO this implementation is not very elegant
if "go" == self.name: if self.name == "go":
res = self.game_engine.executor_do_move(self.history, self.latest_boards, self.board, color, vertex) res = self.game_engine.executor_do_move(self.history, self.latest_boards, self.board, color, vertex)
elif "revsersi" == self.name: elif self.name == "reversi":
res = self.game_engine.executor_do_move(self.board, color, vertex) res = self.game_engine.executor_do_move(self.board, color, vertex)
return res return res

2
AlphaGo/player.py
View File

@ -34,7 +34,7 @@ if __name__ == '__main__':
daemon = Pyro4.Daemon() # make a Pyro daemon daemon = Pyro4.Daemon() # make a Pyro daemon
ns = Pyro4.locateNS() # find the name server ns = Pyro4.locateNS() # find the name server
player = Player(role = args.role, engine = engine) player = Player(role=args.role, engine=engine)
print "Init " + args.role + " player finished" print "Init " + args.role + " player finished"
uri = daemon.register(player) # register the greeting maker as a Pyro object uri = daemon.register(player) # register the greeting maker as a Pyro object
print "Start on name " + args.role print "Start on name " + args.role