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

correct plotting an reloading

Browse Source
This commit is contained in:
steffen-illium
2021-06-07 13:08:41 +02:00
parent 4862407526
commit dbfa97aaba
3 changed files with 18 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
reload_agent.py
View File

@@ -14,21 +14,21 @@ warnings.filterwarnings('ignore', category=UserWarning)
if __name__ == '__main__':
model_name = 'A2C_1622650432'
model_name = 'PPO_1623052687'
run_id = 0
out_path = Path(__file__).parent / 'debug_out'
model_path = out_path / model_name
with Path(r'C:\Users\steff\projects\f_iks\debug_out\yaml.txt').open('r') as f:
env_kwargs = yaml.load(f)
env = SimpleFactory(**env_kwargs)
with (model_path / f'env_{model_name}.yaml').open('r') as f:
env_kwargs = yaml.load(f, Loader=yaml.FullLoader)
env = SimpleFactory(level_name='rooms', **env_kwargs)
# Edit THIS:
model_files = list(natsorted((model_path / f'{run_id}_{model_name}').rglob('*.zip')))
model_files = list(natsorted((model_path / f'{run_id}_{model_name}').rglob('model_*.zip')))
this_model = model_files[0]
model = PPO.load(this_model)
evaluation_result = evaluate_policy(model, env, n_eval_episodes=100, deterministic=False, render=True)
evaluation_result = evaluate_policy(model, env, n_eval_episodes=100, deterministic=True, render=True)
print(evaluation_result)
env.close()