kuhyx/WUT_Computer_Science
1
0
Fork 0
mirror of https://github.com/kuhyx/WUT_Computer_Science.git synced 2026-07-06 11:03:16 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: move hyperparameters before initializaiton of environment

Browse Source
This commit is contained in:
Krzysztof Rudnicki 2023-05-21 19:59:14 +02:00
parent b8522b90bb
commit 135083a5a5
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
main.py
View File

@ -6,7 +6,7 @@ import gymnasium as gym
import numpy as np import numpy as np
def initialize_environment(): def initialize_environment(hyperparameters):
""" """
Initialize environment and video recording Initialize environment and video recording
""" """
@ -18,7 +18,7 @@ def initialize_environment():
env = gym.wrappers.RecordVideo( env = gym.wrappers.RecordVideo(
env, env,
video_folder='vid', video_folder='vid',
episode_trigger=lambda x: x == 1, episode_trigger=lambda x: x == hyperparameters["max_episodes"],
disable_logger=False, disable_logger=False,
name_prefix=time_string) name_prefix=time_string)
return env return env
@ -45,7 +45,7 @@ def initialize_hyperparameters():
"learning_rate": 0.1, "learning_rate": 0.1,
"discount_factor": 0.99, "discount_factor": 0.99,
"epsilon": 0.2, "epsilon": 0.2,
"max_episodes": 1 "max_episodes": 1000
} }
return hyperparameters return hyperparameters
@ -136,9 +136,9 @@ def inference(env, q_table):
if __name__ == '__main__': if __name__ == '__main__':
ENV = initialize_environment()
Q_TABLE = initialize_q_table(ENV)
HYPERPARAMETERS = initialize_hyperparameters() HYPERPARAMETERS = initialize_hyperparameters()
ENV = initialize_environment(HYPERPARAMETERS)
Q_TABLE = initialize_q_table(ENV)
ENV, Q_TABLE = training_loop(HYPERPARAMETERS, ENV, Q_TABLE) ENV, Q_TABLE = training_loop(HYPERPARAMETERS, ENV, Q_TABLE)
inference(ENV, Q_TABLE) inference(ENV, Q_TABLE)