train.create_agent
train.create_agent(
sess,
environment
)
Creates a DQN agent.
Args:
sess
: Atf.Session
object for running associated ops.environment
: An Atari 2600 Gym environment.
Returns:
agent
: An RL agent.
Raises:
ValueError
: Ifagent_name
is not in supported list.