一 tesnsorflow 读取 csv 文件
创建一个 py 文件,文件内容为:
import tensorflow as tf
# 第七个例子 tensorflow 读取 csv 文件
# !!! 注意 csv 文件的格式,有逗号分隔,csv 文件格式错误导致程序一直报错
# 1.导入所需的模块并声明全局变量
tf.compat.v1.disable_eager_execution()
DATA_FILE = 'text1.csv'
# !!! 最后输出结果的行数与该参数有关
BATCH_SIZE = 10
NUM_FEATURES = 14
# 2.定义一个将文件名作为参数的函数,并返回大小等于 BATCH_SIZE 的张量
def data_generator(filename):
"""
:param filename:
:return:
"""
# 3.定义 f_queue 和 reader 为文件名
f_queue = tf.compat.v1.train.string_input_producer(filename)
reader = tf.compat.v1.TextLineReader(skip_header_lines=1)
_, value = reader.read(f_queue)
# 4.这里指定要使用的数据以防数据丢失。对 .csv 解码并选择需要的特征。例如,选择 RM、PTRATIO 和 LSTAT 特征
record_defaults = [[0.0] for _ in range(NUM_FEATURES)]
data = tf.compat.v1.decode_csv(value, record_defaults=record_defaults