# Module: tf_agents.bandits.environments

<br />

|---|
| [![](https://www.tensorflow.org/images/GitHub-Mark-32px.png) View source on GitHub](https://github.com/tensorflow/agents/blob/v0.19.0/tf_agents/bandits/environments/__init__.py) |

Module importing all environments.

## Modules

[`bandit_py_environment`](https://www.tensorflow.org/agents/api_docs/python/tf_agents/bandits/environments/bandit_py_environment) module: Base class for Bandit Python environments.

[`bandit_tf_environment`](https://www.tensorflow.org/agents/api_docs/python/tf_agents/bandits/environments/bandit_tf_environment) module: Base class for bandit environments implemented in TensorFlow.

[`bernoulli_action_mask_tf_environment`](https://www.tensorflow.org/agents/api_docs/python/tf_agents/bandits/environments/bernoulli_action_mask_tf_environment) module: Environment wrapper that adds action masks to a bandit environment.

[`bernoulli_py_environment`](https://www.tensorflow.org/agents/api_docs/python/tf_agents/bandits/environments/bernoulli_py_environment) module: Class implementation of Python Bernoulli Bandit environment.

[`classification_environment`](https://www.tensorflow.org/agents/api_docs/python/tf_agents/bandits/environments/classification_environment) module: An environment based on an arbitrary classification problem.

[`dataset_utilities`](https://www.tensorflow.org/agents/api_docs/python/tf_agents/bandits/environments/dataset_utilities) module: Helper functions for the environments that load datasets.

[`drifting_linear_environment`](https://www.tensorflow.org/agents/api_docs/python/tf_agents/bandits/environments/drifting_linear_environment) module: Bandit drifting linear environment.

[`movielens_per_arm_py_environment`](https://www.tensorflow.org/agents/api_docs/python/tf_agents/bandits/environments/movielens_per_arm_py_environment) module: Class implementation of the per-arm MovieLens Bandit environment.

[`movielens_py_environment`](https://www.tensorflow.org/agents/api_docs/python/tf_agents/bandits/environments/movielens_py_environment) module: Class implementation of the MovieLens Bandit environment.

[`non_stationary_stochastic_environment`](https://www.tensorflow.org/agents/api_docs/python/tf_agents/bandits/environments/non_stationary_stochastic_environment) module: Bandit environment that returns random observations and rewards.

[`piecewise_bernoulli_py_environment`](https://www.tensorflow.org/agents/api_docs/python/tf_agents/bandits/environments/piecewise_bernoulli_py_environment) module: Class implementation of Python Bernoulli Bandit environment.

[`piecewise_stochastic_environment`](https://www.tensorflow.org/agents/api_docs/python/tf_agents/bandits/environments/piecewise_stochastic_environment) module: Bandit piecewise linear stationary environment.

[`random_bandit_environment`](https://www.tensorflow.org/agents/api_docs/python/tf_agents/bandits/environments/random_bandit_environment) module: Bandit environment that returns random observations and rewards.

[`ranking_environment`](https://www.tensorflow.org/agents/api_docs/python/tf_agents/bandits/environments/ranking_environment) module: Ranking Python Bandit environment with items as per-arm features.

[`stationary_stochastic_per_arm_py_environment`](https://www.tensorflow.org/agents/api_docs/python/tf_agents/bandits/environments/stationary_stochastic_per_arm_py_environment) module: Stationary Stochastic Python Bandit environment with per-arm features.

[`stationary_stochastic_py_environment`](https://www.tensorflow.org/agents/api_docs/python/tf_agents/bandits/environments/stationary_stochastic_py_environment) module: Class implementation of Stationary Stochastic Python Bandit environment.

[`stationary_stochastic_structured_py_environment`](https://www.tensorflow.org/agents/api_docs/python/tf_agents/bandits/environments/stationary_stochastic_structured_py_environment) module: Stationary Stochastic Python Bandit environment with structured features.

[`wheel_py_environment`](https://www.tensorflow.org/agents/api_docs/python/tf_agents/bandits/environments/wheel_py_environment) module: Class implementation of Python Wheel Bandit environment.