《Tensorflow机器学习项目实战》Tensorflow 读取Iris数据集CSV格式文件读取

下载并加载数据集

在链接https://vincentarelbundock.github.io/Rdatasets/csv/datasets/iris.csv,下载Iris数据集。Iris数据集是一个包含150个样本的数据集。通俗地说,iris数据集是用来给花做分类的数据集,每个样本包含了花萼长度、花萼宽度、花瓣长度、花瓣宽度四个特征(前4列)。

 Iris数据集CSV格式文件读取:

import tensorflow as tf

sess = tf.Session()
file_name_string = ["iris.csv"]
filename_queue = tf.train.string_input_producer(file_name_string,shuffle=True)
reader = tf.TextLineReader(skip_header_lines=1)
key,value = reader.read(filename_queue)
record_defaults = [[0.],[0.],[0.],[0.],[""]]
col1,col2,col3,col4,col5 = tf.decode_csv(value,record_defaults=record_defaults)
features = tf.stack([col1,col2,col3,col4])

tf.global_variables_initializer().run(session=sess)
tf.local_variables_initializer().run(session=sess)
#tf.initialize_all_variables().run(session=sess)
coord = tf.train.Coordinator()
threads = tf.train.start_queue_runners(coord=coord,sess=sess)

for iteration in range(0,5):
    example = sess.run([features])
    print(example)
    coord.request_stop()
    coord.join(threads)

 输出如图:

猜你喜欢

转载自blog.csdn.net/zqzq19950725/article/details/88168336