python如何读取tfrecord文件_如何将csv文件转换为TFrecord tensorFlow格式？

下面的代码将从多个CSV文件创建TFRecords文件。。。只是还没能读回数据。在import pandas as pd

import numpy as np

import os

import tensorflow as tf

from tqdm import tqdm

def _int64_feature(value):

return tf.train.Feature(int64_list=tf.train.Int64List(value=[value]))

def _float_feature(value):

return tf.train.Feature(float_list=tf.train.FloatList(value=value))

def _bytes_feature(value):

return tf.train.Feature(bytes_list=tf.train.BytesList(value=[value]))

def make_q_list(filepathlist, filetype):

filepathlist = filepathlist

filepaths = []

labels = []

for path in filepathlist:

data_files = os.listdir(path)

for data in data_files:

if data.endswith(filetype):

data_file = os.path.join(path, data)

data_label = os.path.basename(os.path.normpath(path))

filepath