zoukankan      html  css  js  c++  java
  • TensorFlow读取CSV数据

    代码来源于官方文档,做了一些小小的调整:

    # -*- coding:utf-8 -*-
    import tensorflow as tf
    
    filename_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 可以不初始化local
    with 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,1
    10,2,3
    11,3,1
    12,4,2
    17,5,1
    18,6,2
    ......
  • 相关阅读:
    WebSphere--安全性
    WebSphere--会话跟踪
    WebSphere--用户简要表
    WebSphere--连接管理器
    WebSphere--部署Servlet
    WebSphere--定制配置
    WebSphere--安装与配置
    WebSphere--基本特性
    六、Html头部和元信息
    五、Html表单标签
  • 原文地址:https://www.cnblogs.com/hunttown/p/6844461.html
Copyright © 2011-2022 走看看