steffen/marl-factory-grid
1
0
Fork 0
mirror of https://github.com/illiumst/marl-factory-grid.git synced 2025-12-10 17:40:37 +01:00
Code Issues Packages Projects Releases Wiki Activity

added plotting probability maps

Browse Source
This commit is contained in:
Chanumask
2024-05-08 14:27:08 +02:00
parent 39b123221b
commit 0e09094f97
4 changed files with 124 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test_run.py
View File

@@ -8,7 +8,7 @@ from marl_factory_grid.algorithms.static.TSP_item_agent import TSPItemAgent
from marl_factory_grid.algorithms.static.TSP_target_agent import TSPTargetAgent
from marl_factory_grid.environment.factory import Factory
from marl_factory_grid.utils.plotting.plot_single_runs import plot_routes
from marl_factory_grid.utils.plotting.plot_single_runs import plot_routes, plot_action_maps
if __name__ == '__main__':
@@ -41,4 +41,4 @@ if __name__ == '__main__':
print(f'Episode {episode} done...')
break
plot_routes(factory, agents)
plot_action_maps(factory, agents)