날짜를 저장하십시오! Google I / O가 5 월 18 일부터 20 일까지 반환됩니다. 지금 등록

tff.simulation.datasets.build_single_label_dataset

Build a new dataset that only yields examples with a particular label.

This can be used for creating pathological non-iid (in label space) datasets.

dataset the base tf.data.Dataset that yields examples that are structures of string key -> tensor value pairs.
label_key the str key that holds the label for the example.
desired_label the label value to restrict the resulting dataset to.

A tf.data.Dataset that is composed of only examples that have a label matching desired_label.