如何使用Tensorflow加载大型数据集训练深度神经网络?
时间: 2023-04-04 15:04:52 浏览: 107
小型CSV数据.zip
可以使用Tensorflow提供的tf.data API来加载大型数据集,可以使用tf.data.Dataset.from_generator()函数从生成器中加载数据,也可以使用tf.data.TFRecordDataset()函数从TFRecord文件中加载数据。此外,还可以使用tf.data.experimental.CsvDataset()函数从CSV文件中加载数据。在加载数据后,可以使用batch()函数将数据分成小批次进行训练,使用shuffle()函数打乱数据顺序,使用repeat()函数重复数据集以进行多次训练。
阅读全文