From b9961d5645ae1304448e8a7f72dd18390df2ea05 Mon Sep 17 00:00:00 2001 From: DragonDuck24 Date: Mon, 25 May 2026 16:15:23 -0500 Subject: [PATCH] history store --- engine/src/training/train.rs | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/engine/src/training/train.rs b/engine/src/training/train.rs index d6b7bd1..a59deec 100644 --- a/engine/src/training/train.rs +++ b/engine/src/training/train.rs @@ -88,6 +88,8 @@ pub fn train(training_config: TrainingConfig, device: B::Dev let mut board_state = BoardState::default(); let mut episode_buffer: Vec = vec![]; + let mut game_hist = String::new(); + while board_state.status == BoardStateStatus::Ongoing { let results = mcts.search(&board_state, &infer_model, &device); episode_buffer.push(results); @@ -102,9 +104,15 @@ pub fn train(training_config: TrainingConfig, device: B::Dev let stm = board_state.board.side_to_move(); let mv = sample_move(&adjusted, &mut rng, stm).unwrap(); - println!("playing move: {}", mv); + // println!("playing move: {}", mv); + if episode == 0 { + game_hist.push_str(mv.to_string().as_str()); + game_hist.push(' '); + } + board_state.apply_move(mv) } + println!("Game history of first game of iteration: {}", game_hist); for result in episode_buffer.iter().enumerate() { if board_state.status == BoardStateStatus::Stalemate