Fix bug in getBestChild

pbsinclair42 · Feb 27, 2019 · 7403c69 · 7403c69
1 parent bba06d4
commit 7403c69
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/mcts.py b/mcts.py
@@ -95,8 +95,10 @@ def getBestChild(self, node, explorationValue):
         for child in node.children.values():
             nodeValue = child.totalReward / child.numVisits + explorationValue * math.sqrt(
                 2 * math.log(node.numVisits) / child.numVisits)
-            if nodeValue >= bestValue:
+            if nodeValue > bestValue:
                 bestValue = nodeValue
+                bestNodes = [child]
+            elif nodeValue == bestValue:
                 bestNodes.append(child)
         return random.choice(bestNodes)