Upload folder using huggingface_hub

agent.py

@@ -0,0 +1,33 @@
+from stable_baselines3 import A2C
+from tetris_gym.wrappers.observation import ExtendedObservationWrapper
+
+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 = A2C.load("agent/model_tetris")
+        
+        # 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 = ExtendedObservationWrapper(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)
+    

model_tetris.zip

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