Lines Matching refs:dataset
15 """Resampling dataset transformations."""
35 """A transformation that resamples a dataset to achieve a target distribution.
41 class_func: A function mapping an element of the input dataset to a scalar
50 A `Dataset` transformation function, which can be passed to
51 @{tf.data.Dataset.apply}.
54 def _apply_fn(dataset):
55 """Function from `Dataset` to `Dataset` that applies the transformation."""
58 class_values_ds = dataset.map(class_func)
63 initial_dist_ds = dataset_ops.Dataset.from_tensors(
65 acceptance_dist_ds = dataset_ops.Dataset.from_tensors(
100 acceptance_dist_ds = (dataset_ops.Dataset.zip((acceptance_dist_ds,
104 current_probabilities_ds = dataset_ops.Dataset.zip(
107 dataset_ops.Dataset.zip((class_values_ds, current_probabilities_ds,
108 dataset))
