Attend the Women in ML Symposium on December 7 Register now

Module: tf_agents.environments.tf_environment

Stay organized with collections Save and categorize content based on your preferences.

TensorFlow RL Environment API.

Represents a task to be solved, an environment has to define three methods: reset, current_time_step and step.

  • The reset() method returns current time_step after resetting the environment.
  • The current_time_step() method returns current time_step initializing the environmet if needed. Only needed in graph mode.
  • The step(action) method applies the action and returns the new time_step.


class TFEnvironment: Abstract base class for TF RL environments.

absolute_import Instance of __future__._Feature
division Instance of __future__._Feature
print_function Instance of __future__._Feature