Module: tf_agents.bandits.environments.mushroom_environment_utilities

Helper functions for the mushroom dataset.

The dataset is expected to be a CSV with the first column being the label, the other columns represent features. All features, as well as the label, are string features that will be used for one-hot embeddings.


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

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