Module: tf_agents.bandits.environments.dataset_utilities

Helper functions for the environments that load datasets.

Functions

convert_covertype_dataset(...)

convert_mushroom_csv_to_tf_dataset(...): Converts the mushroom CSV dataset into a tf.Dataset.

load_movielens_data(...): Loads the movielens data and returns the ratings matrix.

mushroom_reward_distribution(...): Creates a distribution for rewards for the mushroom environment.

MOVIELENS_NUM_MOVIES 1682
MOVIELENS_NUM_USERS 943