tensorflow1读取csv文件
时间: 2023-11-20 17:51:08 浏览: 97
可以使用 pandas 库来读取 csv 文件,也可以使用 tensorflow 的 Dataset API。下面是使用 tensorflow 读取 csv 文件的示例代码:
```python
import tensorflow as tf
# 定义文件名和默认值
filename = "data.csv"
record_defaults = [tf.float32] * 4
# 创建 Dataset 对象
dataset = tf.data.experimental.CsvDataset(filename, record_defaults, header=True)
# 对数据进行预处理
def preprocess(*cols):
# 将多个列合并为一个张量
features = tf.stack(cols[:-1])
# 将最后一列作为标签
label = cols[-1]
return features, label
# 对每条记录应用 preprocess 函数
dataset = dataset.map(preprocess)
# 创建迭代器
iterator = dataset.make_one_shot_iterator()
# 获取下一条记录
features, label = iterator.get_next()
# 创建会话并运行
with tf.Session() as sess:
while True:
try:
f, l = sess.run([features, label])
print(f, l)
except tf.errors.OutOfRangeError:
break
```
阅读全文