⚠️[FIX] MCTS score

fairyshine · Oct 29, 2024 · f57e07c · f57e07c
1 parent 5efa82c
commit f57e07c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/fastmindapi/algo/tree/mcts.py b/src/fastmindapi/algo/tree/mcts.py
@@ -14,7 +14,7 @@
 
 def calculate_score(node, scalar):
     exploit = node.reward / node.visits
-    explore = math.sqrt(2.0 * math.log(node.visits) / float(node.visits))
+    explore = math.sqrt(2.0 * math.log(node.parent.visits) / float(node.visits))
     return exploit + scalar * explore
 
 class MCTSState(ABC):