本頁面由 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