tensorflow之数据加载

 先上一个示例

参考https://blog.csdn.net/lujiandong1/article/details/53376802

fileDir = 'C:/Users/shenwei/Desktop/data/'
filenames = [fileDir+'A.csv', fileDir+'B.csv', fileDir+'C.csv']
filename_queue = tf.train.string_input_producer(filenames, shuffle=False)
# 定义Reader
reader = tf.TextLineReader()
key, value = reader.read(filename_queue)
# 定义Decoder
example, label = tf.decode_csv(value, record_defaults=[['null'], ['null']])
#example_batch, label_batch = tf.train.shuffle_batch([example,label], batch_size=1, capacity=200, min_after_dequeue=100, num_threads=2)
# 运行Graph
with tf.Session() as sess:
    coord = tf.train.Coordinator()  #创建一个协调器,管理线程
    threads = tf.train.start_queue_runners(coord=coord)  #启动QueueRunner, 此时文件名队列已经进队。
    for i in range(10):
        print (example.eval(),label.eval())
    coord.request_stop()
    coord.join(threads)

猜你喜欢

转载自blog.csdn.net/g0415shenw/article/details/86534870