A transformation that resamples a dataset to achieve a target distribution.
tf.data.experimental.rejection_resample(
class_func, target_dist, initial_dist=None, seed=None
)
Used in the notebooks
Args |
class_func
|
A function mapping an element of the input dataset to a scalar
tf.int32 tensor. Values should be in [0, num_classes) .
|
target_dist
|
A floating point type tensor, shaped [num_classes] .
|
initial_dist
|
(Optional.) A floating point type tensor, shaped
[num_classes] . If not provided, the true class distribution is
estimated live in a streaming fashion.
|
seed
|
(Optional.) Python integer seed for the resampler.
|