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

fixing bug pass parameterg

Browse Source
This commit is contained in:
Wenbo Hu 2017-12-21 19:31:51 +08:00
parent 336cede197
commit 1e2567c174
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tianshou/core/mcts/mcts.py
View File

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