henri/trictrac
1
1
Fork 0
Code Issues 13 Pull requests Projects 1 Releases Packages Wiki Activity Actions

feat: bot all algos

Browse source
This commit is contained in:
Henri Bourcereau 2025-08-21 17:39:45 +02:00
parent 0c58490f87
commit 8f41cc1412
18 changed files with 929 additions and 39 deletions
Download patch file Download diff file Expand all files Collapse all files

6
bot/src/strategy/dqnburn.rs
View file

@ -6,11 +6,11 @@ use crate::{BotStrategy, CheckerMove, Color, GameState, PlayerId};
use log::info;
use store::MoveRules;
use crate::burnrl::dqn_model;
use crate::burnrl::algos::dqn;
use crate::burnrl::environment;
use crate::training_common::{get_valid_action_indices, sample_valid_action, TrictracAction};
type DqnBurnNetwork = dqn_model::Net<NdArray<ElemType>>;
type DqnBurnNetwork = dqn::Net<NdArray<ElemType>>;
/// Stratégie DQN pour le bot - ne fait que charger et utiliser un modèle pré-entraîné
#[derive(Debug)]
@ -40,7 +40,7 @@ impl DqnBurnStrategy {
pub fn new_with_model(model_path: &String) -> Self {
info!("Loading model {model_path:?}");
let mut strategy = Self::new();
strategy.model = dqn_model::load_model(256, model_path);
strategy.model = dqn::load_model(256, model_path);
strategy
}