Создать набор данных из последовательных данных PyTorch - PullRequest
0 голосов
/ 06 апреля 2020

У меня есть набор данных с информацией IMU на каждом временном шаге и соответствующим местоположением истинной наземной точки. Я хочу создать набор данных для подачи в сеть RNN в PyTorch, но я не знаю, как создать класс набора данных, так как необходимо сохранить порядок времени и рассмотреть пакет элементов. Итак, учтите, что у меня есть много CSV-файлов с 4 столбцами (то есть ориентация x, ориентация y, ускорение x, ускорение y) и 10k временными шагами каждый и другие файлы наземных истин с 10k временными шагами и реальным местоположением для каждого временного шага в каждом файле. Я хочу реализовать то, что показано на рисунке ниже:

enter image description here

Я уже выполнил sh, чтобы получить партии IMU из 200 образцов из всех файлов так что размеры [total number of subdivisions or batches, 200, 4]. Теперь я не знаю, как создать набор данных, который будет использоваться PyTorch для итерации по нескольким пакетам одновременно (т. Е. Равен при выполнении Sequence-to-Sequence в тексте, где каждый пакет содержит несколько предложений).

...