Многовариантные временные ряды переменной длины в наборе данных TFRecord - PullRequest
0 голосов
/ 12 ноября 2018

Я пытаюсь составить набор данных TFRecord из репозитория UCI: https://archive.ics.uci.edu/ml/datasets/Character+Trajectories

Данные состоят из 2858 выборок с каждым классом и многомерной (= 3) последовательностью измерений.Максимальная длина составляет 205 шагов.

Когда я создаю набор данных TFRecord, создается впечатление, что tf.train.SequenceExample является объектом, к которому нужно обращаться.Но все примеры охватывают только одномерные временные ряды.

with tf.python_io.TFRecordWriter('/tmp/test.tfrecord') as writer:
for seq, cls in zip(sequences, classes)):
    LEN = seq.shape[1]
    example = tf.train.SequenceExample()
    example.context.feature["length"].int64_list.value.append(LEN)
    example.context.feature["class"].int64_list.value.append(cls)
    example_tokens = record_seq.feature_lists.feature_list['sequence']
    for i in range(LEN):
        #THIS does not work for 3xLEN sequences
        example_tokens.feature.add().float_list.value.append(seq[:,i])

Как записать последовательность 3xLEN в TFrecord, используя SequenceExample?

...