Upload with huggingface_hub

agent.py

@@ -0,0 +1,31 @@
+from stable_baselines3 import PPO
+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 = PPO.load("agent/model_bob")
+        
+        # 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_20x10.zip

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

model_bob.zip

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