B ٻd8@s>dZddlmZddlmZeddedd ddZdS) z#Resampling dataset transformations.) deprecation) tf_exportNz.Use `tf.data.Dataset.rejection_resample(...)`.z$data.experimental.rejection_resamplecsfdd}|S)aA transformation that resamples a dataset to achieve a target distribution. **NOTE** Resampling is performed via rejection sampling; some fraction of the input values will be dropped. 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. Returns: A `Dataset` transformation function, which can be passed to `tf.data.Dataset.apply`. cs|jdS)zEFunction from `Dataset` to `Dataset` that applies the transformation.) class_func target_dist initial_distseed)rejection_resample)Zdataset)rrrrd/var/www/html/venv/lib/python3.7/site-packages/tensorflow/python/data/experimental/ops/resampling.py _apply_fn)s z%rejection_resample.._apply_fnr )rrrrr r )rrrrr rs r)NN)__doc__Ztensorflow.python.utilrZ tensorflow.python.util.tf_exportr deprecatedrr r r r s