Added '--play' flag, so you can now play against the ai.

This commit is contained in:
Alexander Munch-Hansen 2018-05-14 13:07:48 +02:00
parent 2c02689577
commit 00974b0f11
2 changed files with 14 additions and 13 deletions

View File

@ -230,19 +230,17 @@ class Board:
#print("boards after first die: ", boards)
for die in roll[1:]:
# if die != 0:
if True:
# Calculate boards resulting from second move
nested_boards = [calc_moves(board, die) for board in boards]
#print("nested boards: ", nested_boards)
boards = [board for boards in nested_boards for board in boards]
# What the fuck
#for board in boards:
# print(board)
# print("type__:",type(board))
# Add resulting unique boards to set of legal boards resulting from roll
# Calculate boards resulting from second move
nested_boards = [calc_moves(board, die) for board in boards]
#print("nested boards: ", nested_boards)
boards = [board for boards in nested_boards for board in boards]
# What the fuck
#for board in boards:
# print(board)
# print("type__:",type(board))
# Add resulting unique boards to set of legal boards resulting from roll
#print("printing boards from calculate_legal_states: ", boards)
#print("printing boards from calculate_legal_states: ", boards)
legal_moves = legal_moves | set(boards)
# print("legal moves: ", legal_moves)
if len(legal_moves) == 0:

View File

@ -205,7 +205,10 @@ if __name__ == "__main__":
if not config['train_perpetually']:
break
elif args.play:
network = Network(config, config['model'])
network.play_against_network()
elif args.eval:
network = Network(config, config['model'])
for i in range(int(config['repeat_eval'])):