train.create_agent

  1. train.create_agent(
  2. sess,
  3. environment
  4. )

Creates a DQN agent.

Args:

  • sess: A tf.Session object for running associated ops.
  • environment: An Atari 2600 Gym environment.

Returns:

  • agent: An RL agent.

Raises:

  • ValueError: If agent_name is not in supported list.