tensorflow读取csv文件及搭建神经网络训练

1.读取csv文件

import tensorflow as tf  
# 生成一个先入先出队列和一个QueueRunner,生成文件名队列 
def read_batch(filenames,batchsize):
    
    #filenames = ['mer_test/test_set.csv']  
    filename_queue = tf.train.string_input_producer(filenames, shuffle=False)  
    # 定义Reader  
    reader = tf.TextLineReader()  
    key, value = reader.read(filename_queue)  
    # 定义Decoder  
    record_defaults = [[0.0] for _ in range(65)] 
    data= tf.decode_csv(value,record_defaults=record_defaults)  
    
    features=data[1:65]
    #features=(features-tf.reduce_min(features))/(tf.reduce_max(features)-tf.reduce_min(features))
    label=data[0]
    label=tf.cast(label,dtype=tf.uint8)
    label=tf.one_hot(label,2)
    features_batch, label_batch = tf.train.batch([features,label], batch_size=batchsize, capacity=3*batchsize,)  
    return features_batch, label_batch
#测试是否正确读取
if __name__=='__main__':
    filenames = ['test_set.csv'] 
    example_batch, label_batch=read_batch(filenames,2)    
    # 运行Graph
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值