Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

Module: tf_agents.bandits.environments.mushroom_environment_utilities

View source on GitHub

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.

Functions

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.