该
generator参数(可能令人困惑)实际上不应该是一个生成器,而是一个可返回可迭代对象的可调用函数(例如,生成器函数)。可能最简单的选择是使用
lambda。此外,还有一些错误:1)
tf.data.Dataset.from_generator旨在被称为类工厂方法,而不是从实例中调用2)函数(类似于TensorFlow中的其他一些函数)对参数感到奇怪,并且希望您给出序列dtype和每个数据行为
tuples(而不是
listCSV阅读器返回的s)的示例,您可以使用以下示例
map:
import csvimport tensorflow as tfwith open("test.csv", "r") as f: csvreader = csv.reader(f) ds = tf.data.Dataset.from_generator(lambda: map(tuple, csvreader), (tf.uint8,) * (28 ** 2))


