Tensorflow 2.0, как создать оконный набор данных, который сохраняет первые N столбцов постоянными? - PullRequest
0 голосов
/ 01 октября 2019

Я пытаюсь создать оконный набор данных, в котором первые «N» столбцы будут постоянными, а остальные смещены. Например, представьте, что у меня есть пустой массив:

some_data = np.arange(0, 6)
array([0, 1, 2, 3, 4, 5])

Я хотел бы создать набор данных ...

dataset = tf.data.Dataset.from_tensor_slices(some_data)

, который имеет окно со следующими характеристиками:

size = 2
shift = 1
drop_remainder = True
CONSTANT_COLUMNS = 2  # This is what I'm really trying to achieve

Чтобы я мог получить это обратно:

[0 1 2 3]
[0 1 3 4]
[0 1 4 5]

Как видите, первые 2 столбца не меняются. Мой оригинальный подход состоял в том, чтобы создать два набора данных (первый будет содержать только первые 2 столбца, а второй будет содержать фактические окна) и объединить их, но я не смог заставить его работать. Как это сделать? Или может быть есть лучший способ? Спасибо!

...