Tensorflow从CSV创建一个tfrecords文件

您将需要一个单独的脚本将csv文件转换为TFRecords。

假设您有一个带有以下标头的CSV文件：

feature_1, feature_2, ..., feature_n, label

您需要使用读取CSV格式，手动进行

pandas

tf.train.Example

TFRecordWriter

csv = pandas.read_csv("your.csv").valueswith tf.python_io.TFRecordWriter("csv.tfrecords") as writer:    for row in csv:        features, label = row[:-1], row[-1]        example = tf.train.Example()        example.features.feature["features"].float_list.value.extend(features)        example.features.feature["label"].int64_list.value.append(label)        writer.write(example.SerializeToString())