Source code for simple_rl.agents.BeliefAgentClass
# Other imports.
from simple_rl.agents.AgentClass import Agent
from simple_rl.pomdp.BeliefStateClass import BeliefState
[docs]class BeliefAgent(Agent):
def __init__(self, name, actions, gamma=0.99):
'''
Args:
name (str)
actions (list)
gamma (float
'''
Agent.__init__(self, name, actions, gamma)
[docs] def act(self, belief_state, reward):
'''
Args:
belief_state (BeliefState)
reward (float)
Returns:
action (str)
'''
pass
[docs] def policy(self, belief_state):
'''
Args:
belief_state (BeliefState)
Returns:
action (str)
'''
return self.act(belief_state, 0)