s

src/elo.rs

@@ -1,9 +1,9 @@
 use crate::{
-    agent::{Agent, RandomAgent},
+    agent::Agent,
     complexagent::ComplexAgent,
     game_inner::GameInner,
     logic::FutureMoveConfig,
-    repr::{Piece, Winner},
+    repr::{Board, Piece, Winner},
 };
 use rayon::iter::{IntoParallelIterator, ParallelIterator};
 use skillratings::{
@@ -17,16 +17,16 @@ pub fn run() {
         min_arena_depth: 14,
         top_k_children: 2,
         up_to_minus: 10,
-        max_arena_size: 5_000_000,
-        do_not_prune: false,
+        max_arena_size: usize::MAX,
+        do_not_prune: true,
         print: false,
     };
 
     let mut arena = PlayerArena::new(vec![
-        (
-            "RandomAgent".into(),
-            Box::new(|piece| Box::new(RandomAgent::new(piece))),
-        ),
+        // (
+        //     "RandomAgent".into(),
+        //     Box::new(|piece| Box::new(RandomAgent::new(piece))),
+        // ),
         (
             "ComplexAgentD1".into(),
             Box::new(|piece| {
@@ -39,18 +39,6 @@ pub fn run() {
                 ))
             }),
         ),
-        (
-            "ComplexAgentD2".into(),
-            Box::new(|piece| {
-                Box::new(ComplexAgent::new(
-                    piece,
-                    FutureMoveConfig {
-                        max_depth: 2,
-                        ..FMV_BASE
-                    },
-                ))
-            }),
-        ),
         (
             "ComplexAgentD3".into(),
             Box::new(|piece| {
@@ -64,32 +52,46 @@ pub fn run() {
             }),
         ),
         (
-            "ComplexAgentD4".into(),
+            "ComplexAgentD8".into(),
             Box::new(|piece| {
                 Box::new(ComplexAgent::new(
                     piece,
                     FutureMoveConfig {
-                        max_depth: 4,
+                        max_depth: 8,
                         ..FMV_BASE
                     },
                 ))
             }),
         ),
+        // (
+        //     "ComplexAgent5M".into(),
+        //     Box::new(|piece| {
+        //         Box::new(ComplexAgent::new(
+        //             piece,
+        //             FutureMoveConfig {
+        //                 max_arena_size: 5_000_000,
+        //                 ..FMV_BASE
+        //             },
+        //         ))
+        //     }),
+        // ),
     ]);
 
-    arena.play(
-        &(0..arena.players.len())
-            .zip([0].into_iter().cycle())
-            .filter(|(i, j)| i != j)
-            .collect::<Vec<_>>()
-            .repeat(1000),
-    );
-
-    println!("{}", arena);
+    // arena.play(
+    //     &(0..arena.players.len())
+    //         .zip([0].into_iter().cycle())
+    //         .filter(|(i, j)| i != j)
+    //         .collect::<Vec<_>>()
+    //         .repeat(1000),
+    // );
+    for _ in 0..10 {
+        arena.prop_arena(2);
+        println!("{}", arena);
+    }
 }
 
 pub struct PlayerArena {
-    /// Name, Creator, Elo
+    /// Name, Creator Function, Elo
     players: Vec<(String, Box<dyn Fn(Piece) -> Box<dyn Agent>>, EloRating)>,
 }
 
@@ -124,7 +126,7 @@ impl PlayerArena {
 
     fn play(&mut self, pairs: &[(usize, usize)]) {
         pairs
-            .into_iter()
+            .iter()
             .map(|&(i, j)| {
                 (
                     (i, j),
@@ -141,7 +143,7 @@ impl PlayerArena {
             .for_each(|(i, j, o)| self.process_outcome(i, j, &o));
     }
 
-    fn prop_arena(&mut self) {
+    fn prop_arena(&mut self, n: usize) {
         self.play(
             &(0..self.players.len())
                 .flat_map(|i| {
@@ -150,7 +152,8 @@ impl PlayerArena {
                         .filter(|(i, j)| i != j)
                         .collect::<Vec<_>>()
                 })
-                .collect::<Vec<_>>(),
+                .collect::<Vec<_>>()
+                .repeat(n),
         );
     }
 
@@ -173,7 +176,8 @@ impl PlayerArena {
     }
 
     fn play_two_inner(player_1: Box<dyn Agent>, player_2: Box<dyn Agent>) -> Outcomes {
-        let result = GameInner::new(player_1, player_2, false).loop_until_result();
+        let result =
+            GameInner::new(player_1, player_2, false, Board::random(5)).loop_until_result();
 
         match result {
             Winner::Player(piece) => match piece {