# tf_agents.bandits.environments.dataset_utilities.mushroom_reward_distribution

<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/dataset_utilities.py#L108-L140) |

Creates a distribution for rewards for the mushroom environment.

    tf_agents.bandits.environments.dataset_utilities.mushroom_reward_distribution(
        r_noeat, r_eat_safe, r_eat_poison_bad, r_eat_poison_good, prob_poison_bad
    )

<br />

<br />

<br />

| ## Args ||
|---|---|
| `r_noeat` | (float) Reward value for not eating the mushroom. |
| `r_eat_safe` | (float) Reward value for eating an edible mushroom. |
| `r_eat_poison_bad` | (float) Reward value for eating and getting poisoned from a poisonous mushroom. |
| `r_eat_poison_good` | (float) Reward value for surviving after eating a poisonous mushroom. |
| `prob_poison_bad` | Probability of getting poisoned by a poisonous mushroom. |

<br />

<br />

<br />

<br />

| ## Returns ||
|---|---|
| A reward distribution table, instance of `tfd.Distribution`. ||

<br />