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