|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.
convert_mushroom_csv_to_tf_dataset(...): Converts the mushroom CSV dataset into a
mushroom_reward_distribution(...): Creates a distribution for rewards for the mushroom environment.