Python tf.compat.v1.data.TFRecordDataset.rejection_resample用法及代碼示例

用法

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

將數據集重新采樣到目標分布的轉換。

讓我們考慮以下示例，其中初始數據分布為 init_dist 的數據集需要重新采樣到具有 target_dist 分布的數據集。

import collections
initial_dist = [0.5, 0.5]
target_dist = [0.6, 0.4]
num_classes = len(initial_dist)
num_samples = 100000
data_np = np.random.choice(num_classes, num_samples, p=initial_dist)
dataset = tf.data.Dataset.from_tensor_slices(data_np)
x = collections.defaultdict(int)
for i in dataset:
  x[i.numpy()] += 1

根據 initial_dist 分布，x 的值將接近 {0:50000, 1:50000}。

dataset = dataset.rejection_resample(
   class_func=lambda x:x % 2,
   target_dist=target_dist,
   initial_dist=initial_dist)

y = collections.defaultdict(int)
for i in dataset:
  cls, _ = i
  y[cls.numpy()] += 1

y 的值現在將接近 {0:75000, 1:50000} 從而滿足 target_dist 分布。