首页 > 代码库 > TensorFlow读取CSV数据
TensorFlow读取CSV数据
代码来源于官方文档,做了一些小小的调整:
# -*- coding:utf-8 -*-import tensorflow as tffilename_queue = tf.train.string_input_producer(["file01.csv", "file02.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]]col1, col2, col3 = tf.decode_csv(value, record_defaults = record_defaults)features = tf.stack([col1, col2])init_op = tf.global_variables_initializer()local_init_op = tf.local_variables_initializer() # local variables like epoch_num, batch_size 可以不初始化localwith tf.Session() as sess: sess.run(init_op) sess.run(local_init_op) # Start populating the filename queue. coord = tf.train.Coordinator() threads = tf.train.start_queue_runners(coord=coord) for i in range(5): # Retrieve a single instance: example, label = sess.run([features, col3]) print(example) print(label) coord.request_stop() coord.join(threads)
file01.csv 和 file02.csv 格式一样:
19,3,110,2,311,3,112,4,217,5,118,6,2......
TensorFlow读取CSV数据
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。