我用以下函数创建了一个tensorflow数据集：dataset=tf.data.dataset.from_tensor_slices((imgs，labels((。我的标签是"；0"；以及"；1〃；我想把这个数据集分成两个，一个是所有具有标签"的条目；0"；并且一个具有具有标签"0"的所有条目；1〃；。我一直在尝试使用.filter函数来找到一种方法，但没有成功。我们将不胜感激！
Filter将工作。
dataset_zero = dataset.filter(lambda image, label: label == 0)
dataset_one = dataset.filter(lambda image, label: label == 1)
数据集基于流的操作很烦人。如果您的数据集很小，请考虑使用dataset.as_numpy_iterator将数据集检索为列表/numpy数组，并只使用花哨的索引来获取零和一。