我有一个Ubuntu对话语料库的.tfrecords
文件。我正在尝试读取整个数据集,以便将上下文和话语分成批次。使用tf.parse_single_example
,我可以读取单个示例。我尝试使用tf.parse_example
,但我得到以下错误
ValueError: Shape must be rank 1 but is rank 0 for 'ParseExample/ParseExample' (op: 'ParseExample') with input shapes: [], [0], [], [], [], [], [], [0], [0], [0], [0], [0].
我不确定该怎么处理。我使用的代码出现了错误 -
import tensorflow as tf
TRAIN_FILE_TFREC = 'data/train.tfrecords'
filename_queue = tf.train.string_input_producer([TRAIN_FILE_TFREC])
reader = tf.TFRecordReader()
_, serialized_example = reader.read(filename_queue)
features = tf.parse_example(serialized_example,
features = {
"context" : tf.FixedLenFeature([160], tf.int64),
"context_len" : tf.FixedLenFeature([1], tf.int64),
"utterance" : tf.FixedLenFeature([80], tf.int64),
"utterance_len" : tf.FixedLenFeature([1], tf.int64),
"label" : tf.FixedLenFeature([1], tf.int64)
})
任何想法。
parse_single_example
起作用了,但你的代码使用了parse_single_example
。你能展示给我们出错的精确代码吗? - Peter Hawkinsparse_single_example
而不是parse_example
。 - Clock Slave