I have a .tfrecords
file of the Ubuntu Dialog Corpus. I am trying to read in the whole dataset so that I can split the contexts and utterances into batches. Using tf.parse_single_example
I was able to read in a single example. I tried using tf.parse_example
but I get the following error
ValueError: Shape must be rank 1 but is rank 0 for 'ParseExample/ParseExample' (op: 'ParseExample') with input shapes: [], [0], [], [], [], [], [], [0], [0], [0], [0], [0].
I am not sure what to make of it. The code I used to get to the error -
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)
})
Any ideas