def play(self, endtime): root = MCTSNode(self.propnet) for i in range(500): simulation(root) root.print_node() best, choice = -1, None for i, c in root.move_counts[self.role].items(): if c > best: best, choice = c, i move = self.propnet.id_to_move[choice].move_gdl print('Made move', move) return move
#!/usr/bin/env python3.7 import sys sys.path.insert(1, '/Users/Cameron/Desktop/transfer_ggp') from model import Model from mcts import MCTSNode, simulation from propnet.propnet import load_propnet import time start = time.time() # propnet = load_propnet('connect4match1') # propnet = load_propnet('tictactoe1') data, propnet = load_propnet('connectFour') root = MCTSNode(propnet, data) # exit(0) for i in range(400): simulation(root) root.print_node() print('Took', time.time() - start, 'seconds')