From f820aab0081dff11d5ffe7d29914a27c0ae2c40b Mon Sep 17 00:00:00 2001 From: Wenbo Hu Date: Tue, 12 Dec 2017 20:37:57 +0800 Subject: [PATCH] change mcts steps --- AlphaGo/strategy.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/AlphaGo/strategy.py b/AlphaGo/strategy.py index ff5d79f..327111d 100644 --- a/AlphaGo/strategy.py +++ b/AlphaGo/strategy.py @@ -246,7 +246,7 @@ class strategy(object): self.simulator.history = copy.copy(history) self.simulator.board = copy.copy(history[-1]) state = self.data_process(self.simulator.history, color) - mcts = MCTS(self.simulator, self.evaluator, state, self.simulator.size ** 2 + 1, inverse=True, max_step=100) + mcts = MCTS(self.simulator, self.evaluator, state, self.simulator.size ** 2 + 1, inverse=True, max_step=10) temp = 1 prob = mcts.root.N ** temp / np.sum(mcts.root.N ** temp) choice = np.random.choice(self.simulator.size ** 2 + 1, 1, p=prob).tolist()[0]