steffen/marl-factory-grid
1
0
Fork 0
mirror of https://github.com/illiumst/marl-factory-grid.git synced 2025-05-23 07:16:44 +02:00
Code Issues Packages Projects Releases Wiki Activity

removed test script from main

Browse Source
This commit is contained in:
romue 2021-06-02 14:02:17 +02:00
parent 1425f75643
commit 8810955e86
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
main.py
View File

@ -41,6 +41,8 @@ def combine_runs(run_path: Union[str, PathLike]):
value_vars=columns, var_name="Measurement",
value_name="Score")
df_melted = df_melted[df_melted['Episode'] % skip_n == 0]
#df_melted['Episode'] = df_melted['Episode'] * skip_n # only needed for old version
prepare_plot(run_path / f'{run_path.name}_monitor_lineplot.png', df_melted)
print('Plotting done.')
@ -49,16 +51,12 @@ def combine_runs(run_path: Union[str, PathLike]):
if __name__ == '__main__':
from stable_baselines3 import PPO, DQN, A2C
from algorithms.dqn_reg import RegDQN
dirt_props = DirtProperties()
time_stamp = int(time.time())
out_path = None
combine_runs(Path(__file__).parent / 'debug_out'/ 'A2C_1622571986')
exit()
for modeL_type in [RegDQN, DQN]:
for modeL_type in [A2C, PPO, DQN]:
for seed in range(5):
env = SimpleFactory(n_agents=1, dirt_properties=dirt_props, pomdp_radius=2, max_steps=400,