File size: 556 Bytes
0c51b93 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
import abc
class Agent(object):
def reset(self):
"""For state-full agents this function performs reseting at the beginning of each episode."""
pass
@abc.abstractmethod
def train(self, training=True):
"""Sets the agent in either training or evaluation mode."""
@abc.abstractmethod
def update(self, replay_buffer, logger, step):
"""Main function of the agent that performs learning."""
@abc.abstractmethod
def act(self, obs, sample=False):
"""Issues an action given an observation."""
|