tensorflow将CSV文件转为TFrecords文件

'''
将CSV文件数据转为TFrecords文件格式
'''
import timeit   #查看运行开始到结束所用的时间
import tensorflow as tf
import os

def generate_tfrecords(input_filename, output_filename):
    print("\nStart to convert {} to {}\n".format(input_filename, output_filename))
    start_time = timeit.default_timer()
    writer = tf.python_io.TFRecordWriter(output_filename)

    for line in open(input_filename, "r"):
        data = line.split(",")
        label = float(data[9])
        
       
        features = [float(i) for i in data[:9]]   #特征不要最后一列数据
        #将数据转化为原生 bytes
        example = tf.train.Example(features=tf.train.Features(feature={
            "label":
                tf.train.Fe
  • 1
    点赞
  • 14
    收藏
    觉得还不错? 一键收藏
  • 15
    评论
评论 15
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值