转，读csv生成tf-record格式

1. import timeit #查看运行开始到结束所用的时间
2. import tensorflow as tf
3. import os
5. def generate_tfrecords(input_filename, output_filename):
6. print( "\nStart to convert {} to {}\n".format(input_filename, output_filename))
7. start_time = timeit.default_timer()
8. writer = tf.python_io.TFRecordWriter(output_filename)
10. for line in open(input_filename, "r"):
11. data = line.split( ",")
12. label = float(data[ 9])
15. features = [float(i) for i in data[: 9]] #特征不要最后一列数据
16. #将数据转化为原生 bytes
17. example = tf.train.Example(features=tf.train.Features(feature={
18. "label":
19. tf.train.Feature(float_list=tf.train.FloatList(value=[label])),
20. "features":
21. tf.train.Feature(float_list=tf.train.FloatList(value=features)),
22. }))
23. writer.write(example.SerializeToString()) #序列化为字符串
25. writer.close()
26. print( "Successfully convert {} to {}".format(input_filename,
27. output_filename))
28. end_time = timeit.default_timer()
29. print( "\nThe pretraining process ran for {0} minutes\n".format((end_time - start_time) / 60))
31. def main():
32. current_path = "G:/Spyder/csv_TFrecords/"
33. for filename in os.listdir(current_path):
34. if filename == "SiteB.UDP_train_unsupervised.csv": #当前路径下，需要转换的CSV文件
35. generate_tfrecords(current_path+filename, current_path+filename + ".tfrecords")
37. return current_path+filename + ".tfrecords"
39. if __name__ == "__main__":

1. import timeit #查看运行开始到结束所用的时间
2. import tensorflow as tf
3. import os
5. def generate_tfrecords(input_filename, output_filename):
6. print( "\nStart to convert {} to {}\n".format(input_filename, output_filename))
7. start_time = timeit.default_timer()
8. writer = tf.python_io.TFRecordWriter(output_filename)
10. for line in open(input_filename, "r"):
11. data = line.split( ",")
12. label = float(data[ 9])
15. features = [float(i) for i in data[: 9]] #特征不要最后一列数据
16. #将数据转化为原生 bytes
17. example = tf.train.Example(features=tf.train.Features(feature={
18. "label":
19. tf.train.Feature(float_list=tf.train.FloatList(value=[label])),
20. "features":
21. tf.train.Feature(float_list=tf.train.FloatList(value=features)),
22. }))
23. writer.write(example.SerializeToString()) #序列化为字符串
25. writer.close()
26. print( "Successfully convert {} to {}".format(input_filename,
27. output_filename))
28. end_time = timeit.default_timer()
29. print( "\nThe pretraining process ran for {0} minutes\n".format((end_time - start_time) / 60))
31. def main():
32. current_path = "G:/Spyder/csv_TFrecords/"
33. for filename in os.listdir(current_path):
34. if filename == "SiteB.UDP_train_unsupervised.csv": #当前路径下，需要转换的CSV文件
35. generate_tfrecords(current_path+filename, current_path+filename + ".tfrecords")
37. return current_path+filename + ".tfrecords"
39. if __name__ == "__main__":