dacbench.agents.generic_agent

Module Contents

Classes

GenericAgent

Abstract class to implement for use with the runner function

class dacbench.agents.generic_agent.GenericAgent(env, policy)

Bases: dacbench.abstract_agent.AbstractDACBenchAgent

Abstract class to implement for use with the runner function

act(self, state, reward)

Compute and return environment action

Parameters
  • state – Environment state

  • reward – Environment reward

Returns

Action to take

Return type

action

train(self, next_state, reward)

Train during episode if needed (pass if not)

Parameters
  • next_state – Environment state after step

  • reward – Environment reward

end_episode(self, state, reward)

End of episode training if needed (pass if not)

Parameters
  • state – Environment state

  • reward – Environment reward