TensorFlow如何实现从CSV文件中读取数据的示例？
- 内容介绍
- 文章标签
- 相关推荐
本文共计2386个文字，预计阅读时间需要10分钟。
TensorFlow 读取CSV数据原理简介，直接通过代码实现：
pythonimport tensorflow as tf
方法一：详细读取tf_read.csv代码def read_csv(): # 定义CSV文件路径 file_path='tf_read.csv'
# 创建一个读取CSV文件的tf.data.Dataset对象 dataset=tf.data.experimental.make_csv_dataset( file_path, column_names=['col1', 'col2', 'col3'], # 指定列名 batch_size=10, # 每批读取的数据量 label_name='label' # 指定标签列名 )
# 打印数据 for batch in dataset.take(1): # 取一个批次的数据进行打印 print(batch)
运行上述代码即可读取CSV文件并打印数据。
本文共计2386个文字，预计阅读时间需要10分钟。
TensorFlow 读取CSV数据原理简介，直接通过代码实现：
pythonimport tensorflow as tf
方法一：详细读取tf_read.csv代码def read_csv(): # 定义CSV文件路径 file_path='tf_read.csv'
# 创建一个读取CSV文件的tf.data.Dataset对象 dataset=tf.data.experimental.make_csv_dataset( file_path, column_names=['col1', 'col2', 'col3'], # 指定列名 batch_size=10, # 每批读取的数据量 label_name='label' # 指定标签列名 )
# 打印数据 for batch in dataset.take(1): # 取一个批次的数据进行打印 print(batch)
运行上述代码即可读取CSV文件并打印数据。