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