steffen/marl-factory-grid
1
0
Fork 0
mirror of https://github.com/illiumst/marl-factory-grid.git synced 2025-11-27 11:30:38 +01:00
Code Issues Packages Projects Releases Wiki Activity

added own dqn

Browse Source
This commit is contained in:
romue
2021-06-16 23:19:01 +02:00
parent 67f7f3500c
commit 0a5a958de2
2 changed files with 59 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
environments/factory/base_factory.py
View File

@@ -23,6 +23,7 @@ class BaseFactory(gym.Env):
@property
def observation_space(self):
agent_slice = self.n_agents if self.omit_agent_slice_in_obs else 0
agent_slice = 1 if self.combin_agent_slices_in_obs else agent_slice
if self.pomdp_radius:
return spaces.Box(low=0, high=1, shape=(self._state.shape[0] - agent_slice, self.pomdp_radius * 2 + 1,
self.pomdp_radius * 2 + 1), dtype=np.float32)