python 深度学习之tensorflow v2.3（四）

一 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