From 2dac9c71f39c5bdc00ad44832ccaf7dce86569b7 Mon Sep 17 00:00:00 2001 From: Tue Herlau <tuhe@dtu.dk> Date: Tue, 18 Feb 2025 10:55:42 +0100 Subject: [PATCH] Updated inventory control example --- .../lec02/lecture_02_optimal_inventory.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) create mode 100644 irlc/lectures/lec02/lecture_02_optimal_inventory.py diff --git a/irlc/lectures/lec02/lecture_02_optimal_inventory.py b/irlc/lectures/lec02/lecture_02_optimal_inventory.py new file mode 100644 index 0000000..48eb5e1 --- /dev/null +++ b/irlc/lectures/lec02/lecture_02_optimal_inventory.py @@ -0,0 +1,15 @@ +# This file may not be shared/redistributed without permission. Please read copyright notice in the git repo. If this file contains other copyright notices disregard this text. +from irlc import Agent, interactive +from irlc.lectures.lec01.viz_inventory_environment import VizInventoryEnvironment + +if __name__ == "__main__": + env = VizInventoryEnvironment(render_mode='human') + from irlc.ex02.inventory import InventoryDPModel + from irlc.ex02.dp_agent import DynamicalProgrammingAgent + + from irlc import train, interactive + agent = DynamicalProgrammingAgent(env, model=InventoryDPModel()) + + env, agent = interactive(env, agent) + n = 400 + stats, _ = train(env, agent, max_steps=n, num_episodes=1000, return_trajectory=False, verbose=False) -- GitLab