此页面由 Cloud Translation API 翻译。
Switch to English

tff.simulation.datasets.build_single_label_dataset

GitHub上查看源代码

建立一个新的数据集,只有获得了具有特定标签的例子。

这可以用于创建病理非IID(在标签空间)的数据集。

dataset 基座tf.data.Dataset其产量的例子是字符串键的结构- >张量值对。
label_key str是保存了例如标签密钥。
desired_label 标签值,以限制所产生的数据集。

tf.data.Dataset其由仅具有一个标签匹配例子desired_label