tensorflow随笔-文件数据读取

「已注销」

已于 2022-12-12 21:41:35 修改

阅读量1.1k

点赞数

文章标签： tensorflow

于 2019-04-25 11:56:05 首次发布

本文链接：https://blog.csdn.net/AI_LX/article/details/89513228

版权

# -*- coding:utf-8 -*-
import tensorflow as tf
fn_queue=tf.train.string_input_producer(["winequality-white-test.csv"])
reader=tf.TextLineReader(skip_header_lines=1)
key,value=reader.read(fn_queue)
print(key)#文件名及行数
print(value)#按行读取的原始字符串
#fixedacidity,volatileacidity,citric acid,residualsugar,chlorides,freesulfurdioxide,totalsulfurdioxide,density,pH,sulphates,alcohol,quality
recordDefaults=[[1.],[1.],[1.],[1.],[1.],[1.],[1.],[1.],[1.],[1.],[1.],[1.]]
(fixedacidity,volatileacidity,citricacid,residualsugar,chlorides,freesulfurdioxide,totalsulfurdioxide,density,pH,sulphates,alcohol,quality)=\
tf.decode_csv(value,record_defaults=recordDefaults)  
features=tf.stack([fixedacidity,volatileacidity,citricacid,residualsugar,chlorides,freesulfurdioxide,totalsulfurdioxide,density,pH,sulphates,alcohol])
out=quality
init_op=tf.global_variables_initializer()

with tf.Session() as sess:
    sess.run(init_op)
    #开启协调器
    coord=tf.train.Coordinator()
    #启动队列填充
    threads=tf.train.start_queue_runners(coord=coord)
    try:
        for i in range(200):
            sample,label=sess.run([features,out])
            print(sample)
            print(label)
            print("--------")
    except tf.errors.OutOfRangeError:
        print("Done!!!")
    finally:
        coord.request_stop()
        coord.join(threads)
        print("reading has finished.")

以白酒品质样本文件为例，输出结果如下：
…
…
[6.100e+00 2.700e-01 4.300e-01 7.500e+00 4.900e-02 6.500e+01 2.430e+02
9.957e-01 3.120e+00 4.700e-01 9.000e+00]
5.0
reading has finished.