Add a DQN example to the reinforcement-learning section (#1872)

This commit is contained in:
Gabriel
2024-03-18 21:22:53 +01:00
committed by GitHub
parent 04a61a9c72
commit 6a966cf9e0
2 changed files with 121 additions and 0 deletions

View File

@ -14,6 +14,7 @@ mod vec_gym_env;
mod ddpg;
mod policy_gradient;
mod dqn;
#[derive(Parser)]
struct Args {
@ -25,6 +26,7 @@ struct Args {
enum Command {
Pg,
Ddpg,
Dqn
}
fn main() -> Result<()> {
@ -32,6 +34,7 @@ fn main() -> Result<()> {
match args.command {
Command::Pg => policy_gradient::run()?,
Command::Ddpg => ddpg::run()?,
Command::Dqn => dqn::run()?
}
Ok(())
}