Module: tf_agents.bandits.environments

Module importing all environments.

Modules

bandit_py_environment module: Base class for Bandit Python environments.

bandit_tf_environment module: Base class for bandit environments implemented in TensorFlow.

bernoulli_action_mask_tf_environment module: Environment wrapper that adds action masks to a bandit environment.

bernoulli_py_environment module: Class implementation of Python Bernoulli Bandit environment.

classification_environment module: An environment based on an arbitrary classification problem.

dataset_utilities module: Helper functions for the environments that load datasets.

drifting_linear_environment module: Bandit drifting linear environment.

movielens_per_arm_py_environment module: Class implementation of the per-arm MovieLens Bandit environment.

movielens_py_environment module: Class implementation of the MovieLens Bandit environment.

non_stationary_stochastic_environment module: Bandit environment that returns random observations and rewards.

piecewise_bernoulli_py_environment module: Class implementation of Python Bernoulli Bandit environment.

piecewise_stochastic_environment module: Bandit piecewise linear stationary environment.

random_bandit_environment module: Bandit environment that returns random observations and rewards.

ranking_environment module: Ranking Python Bandit environment with items as per-arm features.

stationary_stochastic_per_arm_py_environment module: Stationary Stochastic Python Bandit environment with per-arm features.

stationary_stochastic_py_environment module: Class implementation of Stationary Stochastic Python Bandit environment.

stationary_stochastic_structured_py_environment module: Stationary Stochastic Python Bandit environment with structured features.

wheel_py_environment module: Class implementation of Python Wheel Bandit environment.