инструкция по обновлению устаревшего скользящего окна набора данных с помощью tf.data.experimental.CsvDataset - PullRequest
1 голос
/ 23 марта 2019

Я безуспешно пытаюсь обновить свой код с помощью инструкции, приведенной в документации по тензорному потоку (api r1.13).Я использую tf.data.experimental.CsvDataset и устаревший tf.contrib.data.sliding_window_batch для RNN, и все работает нормально (кроме устаревшего предупреждающего сообщения slide_window).

Для обновления я просто заменил

dataset = dataset.apply(tf.contrib.data.sliding_window_batch(batch_size, 1))

на

dataset = dataset.window(size=batch_size, stride=1).flat_map(lambda x: x.batch(batch_size))

, и я получил следующую ошибку для CSV-файла с 50 столбцами:

TypeError: <lambda>() takes 1 positional argument but 50 were given

Как я могу решить эту проблему для любого файла CSV (с любым количеством столбцов)?

...