Searched refs:class_values_ds (Results 1 – 1 of 1) sorted by relevance
178 target_dist_t, class_values_ds, dist_estimation_batch_size=32, argument191 initial_dist_ds = (class_values_ds.batch(dist_estimation_batch_size)