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

Item und dirt funktionieren nun zusammen

Browse Source 
names refactored
This commit is contained in:
Steffen Illium
2021-09-07 16:04:04 +02:00
parent d0e1175ff1
commit 444ffe3f37
9 changed files with 121 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
algorithms/q_learner.py
View File

@ -99,7 +99,7 @@ class QLearner(BaseLearner):
if __name__ == '__main__':
from environments.factory.simple_factory import SimpleFactory, DirtProperties, MovementProperties
from environments.factory.factory_dirt import DirtFactory, DirtProperties, MovementProperties
from algorithms.common import BaseDDQN, BaseICM
from algorithms.m_q_learner import MQLearner, MQICMLearner
from algorithms.vdn_learner import VDNLearner
@ -109,7 +109,7 @@ if __name__ == '__main__':
with (Path(f'../environments/factory/env_default_param.yaml')).open('r') as f:
env_kwargs = yaml.load(f, Loader=yaml.FullLoader)
env = SimpleFactory(**env_kwargs)
env = DirtFactory(**env_kwargs)
obs_shape = np.prod(env.observation_space.shape)
n_actions = env.action_space.n