File size: 1,221 Bytes
302a5d2 f8dc19d 302a5d2 6322b9f 302a5d2 100d782 302a5d2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
from stable_baselines3 import A2C
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 = A2C.load("agent/Sokoban-v1_5_8_1box_A2C")
# 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(observation, deterministic=True)
|