bot train graph
This commit is contained in:
parent
c0d42a0c45
commit
744a70cf1d
3 changed files with 7 additions and 5 deletions
|
|
@ -91,7 +91,7 @@ impl Environment for TrictracEnvironment {
|
|||
type ActionType = TrictracAction;
|
||||
type RewardType = f32;
|
||||
|
||||
const MAX_STEPS: usize = 1000; // Limite max pour éviter les parties infinies
|
||||
const MAX_STEPS: usize = 700; // Limite max pour éviter les parties infinies
|
||||
|
||||
fn new(visualized: bool) -> Self {
|
||||
let mut game = GameState::new(false);
|
||||
|
|
|
|||
|
|
@ -9,7 +9,7 @@ type Backend = Autodiff<NdArray<ElemType>>;
|
|||
type Env = environment::TrictracEnvironment;
|
||||
|
||||
fn main() {
|
||||
println!("> Entraînement");
|
||||
// println!("> Entraînement");
|
||||
let conf = dqn_model::DqnConfig {
|
||||
num_episodes: 40,
|
||||
// memory_size: 8192, // must be set in dqn_model.rs with the MEMORY_SIZE constant
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue