seems like avg reward performs better with our relatively low iter count

jackschu · Oct 22, 2023 · 3c4ead3 · 3c4ead3
1 parent 7a0f4a4
commit 3c4ead3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/bots/mcts_bot.rs b/src/bots/mcts_bot.rs
@@ -114,8 +114,8 @@ impl<'a> ArenaTree<'a> {
             .children
             .iter()
             .max_by(|l, r| {
-                let lv = self.arena[*l.to_owned()].data.visited;
-                let rv = self.arena[*r.to_owned()].data.visited;
+                let lv = self.arena[*l.to_owned()].data.average_reward();
+                let rv = self.arena[*r.to_owned()].data.average_reward();
                 lv.partial_cmp(&rv).unwrap()
             })
             .unwrap();