diff --git a/src/elo.rs b/src/elo.rs index 18e45f5..1fefe4e 100644 --- a/src/elo.rs +++ b/src/elo.rs @@ -25,7 +25,7 @@ pub fn run() { children_eval_method: ChildrenEvalMethod::Average, }; - let configs = (1..=6) + let configs = (4..=6) .map(move |d| FutureMoveConfig { max_depth: d, ..FMV_BASE @@ -51,6 +51,23 @@ pub fn run() { ..prev_c }) .to_vec() + }) + .flat_map(move |prev_c| { + if prev_c.do_not_prune { + // do not bother making configs where do_not_prune is true + // as top_k_children does nothing when pruning is skipped + return vec![prev_c]; + } + + // different values of top_k_children + [1, 2, 3] + .into_iter() + .filter(|&x| x <= prev_c.max_depth) + .map(move |ad_offset| FutureMoveConfig { + min_arena_depth: prev_c.max_depth - ad_offset, + ..prev_c + }) + .collect() }); let vec: Vec<(String, Box Box>)> = configs