首页 > 解决方案 > 张量流中 tf.decode_csv 的记录默认值

问题描述

我在 tensorflow 中使用 tf.decode_csv 作为解码器,将制表符分隔文件中的训练示例解析为 cnn 模型。对于每个训练示例,特征都是二维的(100 列,2000 行)。看完tensorflow官网的文档后,我还有两个疑问。

  1. 如何创建record_defaults?以下是我的代码,但我不确定它是否正确。

代码

filename_queue = tf.train.string_input_producer([file], num_epochs)

key, value = tf.TextLineReader().read(filename_queue)

record_defaults = [[1.0 for col in range(0, 100)] for row in range(0, 2000)]

content = tf.decode_csv(value, record_defaults = record_defaults, field_delim = '\t')

features = tf.pack(content[0:1999])
  1. 我正在做二进制 (0, 1) 分类。我在哪里放置训练示例的标签?在第 2001 行?(对于每个训练示例,前 2000 行是特征,第 2001 行是标签)

谢谢你的时间!

标签: tensorflow

解决方案


推荐阅读