tff.simulation.datasets.build_dataset_mixture

Build a new dataset that probabilistically returns examples.

tff.simulation.datasets.build_dataset_mixture(
    a, b, a_probability, op_seed=None
)

Args:

a: the first tf.data.Dataset.
b: the second tf.data.Dataset.
a_probability: the float probability to select the next example from the a dataset.
op_seed: an optional int seed for the TensorFlow PRNG op. Strongly recommended to only use in unittests. Note: only setting this seed will not enable deterministic behavior, callers must also use tf.random.set_seed to enable deterministic behavior.

Returns:

A tf.data.Dataset that returns examples from dataset a with probability a_probability, and examples form dataset b with probability (1 - a_probability). The dataset will yield the number of examples equal to the smaller of a or b.