minor revision.

This commit is contained in:
Wenbo Hu 2017-12-20 21:52:30 +08:00
parent 8875ad1bf7
commit 0ab38743aa
2 changed files with 2 additions and 2 deletions

View File

@ -180,7 +180,6 @@ class Go:
if not self._knowledge_prunning(current_board, color, vertex):
return False
return True
def simulate_is_valid_list(self, state, action_set):
@ -188,7 +187,7 @@ class Go:
## if no action is valid, then pass
valid_action_list = []
for action_candidate in action_set:
if self.simulate_is_valid(state, action_candidate):
if not self.simulate_is_valid(state, action_candidate):
valid_action_list.append(action_candidate)
if not valid_action_list:
valid_action_list.append(utils.PASS)

View File

@ -71,6 +71,7 @@ class UCTNode(MCTSNode):
self.parent.backpropagation(self.children[action].reward)
def valid_mask(self, simulator):
# let all invalid actions illeagel in mcts
if self.mask is None:
self.mask = simulator.simulate_is_valid_list(self.state, range(self.action_num - 1))
self.ucb[self.mask] = -float("Inf")