Upload folder using huggingface_hub

Sokoban-A2C-CNN5.zip

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8983aa53b7414466ea1c1855b530827bd71528f0f7aa64a70a750844992e6c4c
-size 15276121
+oid sha256:cee537a4dafdf298a59b47718fb12f3291dbac6bea11d0839d1fa0711ee61fa1
+size 15277144

Sokoban-A2C-CNN6.zip

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cd6959441e213b244801a02ebd3cedbcaaee70fa11fdd2089bbf479f770f244c
+size 15277144

Sokoban-A2C-CNN7.zip

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:704c789fd5a0fdedf3f1af1fe94aa367f748831edd6e7b824e59cce261a7775e
+size 15274932

agent.py

@@ -1,33 +1,33 @@
-from stable_baselines3 import PPO
-from sokoban_gym.wrappers.observation import ImageObservationWrapper
-
-class Agent:
-    """
-    A kötelező programként beadandó ágens leírása.
-    """
-
-    def __init__(self, env) -> None:
-        """
-        A konsztruktorban van lehetőség például a modell betöltésére
-        vagy a környezet wrapper-ekkel való kiterjesztésére.
-        """
-        
-        self.model = PPO.load("agent/Sokoban-A2C-CNN5")
-        
-        # A környezetet kiterjeszthetjük wrapper-ek segítségével.
-        # Ha tanításkor modosítottuk a megfigyeléseket,
-        # akkor azt a módosítást kiértékeléskor is meg kell adnunk.
-        self.observation_wrapper = ImageObservationWrapper(env)
-
-    def act(self, observation):
-        """
-        A megfigyelés alapján visszaadja a következő lépést.
-        Ez a függvény fogja megadni az ágens működését.
-        """
-
-        # Ha tanításkor modosítottuk a megfigyeléseket,
-        # akkor azt a módosítást kiértékeléskor is meg kell adnunk.
-        extended_obsetvation = self.observation_wrapper.observation(observation)
-
-        return self.model.predict(extended_obsetvation, deterministic=True)
+from stable_baselines3 import PPO
+from sokoban_gym.wrappers.observation import ImageObservationWrapper
+
+class Agent:
+    """
+    A kötelező programként beadandó ágens leírása.
+    """
+
+    def __init__(self, env) -> None:
+        """
+        A konsztruktorban van lehetőség például a modell betöltésére
+        vagy a környezet wrapper-ekkel való kiterjesztésére.
+        """
+        
+        self.model = PPO.load("agent/Sokoban-A2C-CNN7")
+        
+        # A környezetet kiterjeszthetjük wrapper-ek segítségével.
+        # Ha tanításkor modosítottuk a megfigyeléseket,
+        # akkor azt a módosítást kiértékeléskor is meg kell adnunk.
+        self.observation_wrapper = ImageObservationWrapper(env)
+
+    def act(self, observation):
+        """
+        A megfigyelés alapján visszaadja a következő lépést.
+        Ez a függvény fogja megadni az ágens működését.
+        """
+
+        # Ha tanításkor modosítottuk a megfigyeléseket,
+        # akkor azt a módosítást kiértékeléskor is meg kell adnunk.
+        extended_obsetvation = self.observation_wrapper.observation(observation)
+
+        return self.model.predict(extended_obsetvation, deterministic=True)