tensorflow读取csv文件

 

import tensorflow as tf
filename_queue = tf.train.string_input_producer(["file0.csv","file1.csv"])

reader = tf.TextLineReader()
key, value = reader.read(filename_queue)

# Default values, in case of empty columns. Also specifies the type of the
# decoded result.

record_defaults = [[1], [1], [1], [1], [1]]
col1, col2, col3, col4, col5 = tf.decode_csv(
value, record_defaults=record_defaults)

features = tf.stack([col1, col2, col3, col4],0)

with tf.Session() as sess:
# Start populating the filename queue.
coord = tf.train.Coordinator()
threads = tf.train.start_queue_runners(coord=coord)

for i in range(10):

# Retrieve a single instance:
example, label = sess.run([features, col5])
print(example,'::',label,'::')

coord.request_stop()
coord.join(threads)