TensorFlow對(duì)數(shù)據(jù)集重新取樣

tf.contrib.data.rejection_resample

rejection_resample(
    dataset,
    class_func,
    target_dist,
    initial_dist=None,
    seed=None
)

定義在：tensorflow/contrib/data/python/ops/dataset_ops.py.

重新取樣此數(shù)據(jù)集以實(shí)現(xiàn)目標(biāo)類分布.

注意：重新取樣是通過(guò)拒絕抽樣進(jìn)行的；輸入值的一部分將被刪除.

ARGS：

• dataset：一個(gè)數(shù)據(jù)集對(duì)象.
• class_func：一個(gè)函數(shù),用于映射張量的嵌套結(jié)構(gòu)(具有由 dataset.output_shapes 和 dataset.output_types 定義的形狀和類型)到標(biāo)量 tf.int32 的張量,值應(yīng)在[0, num_classes].
• target_dist：浮點(diǎn)型張量,形狀為[num_classes].
• initial_dist：(可選)浮點(diǎn)型張量,形狀為[num_classes].如果沒(méi)有提供,那么真實(shí)的類分布是以流式方式進(jìn)行估計(jì)的.
• seed：(可選)用于重新取樣的 Python 整數(shù)種子.

返回：

返回一個(gè)數(shù)據(jù)集.