0

假设我使用 keras 创建了以下时间序列生成器:

from tensorflow.keras.preprocessing.sequence import TimeseriesGenerator
gen = TimeseriesGenerator(data=[1,2,3,4,5,6,7,8,9], targets=[1,2,3,4,5,6,7,8,9], length=2, batch_size=1, start_index=5)

由于它具有设置start_index=5,它将跳过前 5 个数据点,因此gen仅包含这些实际可用的数据:

# first are the data points [x_n, x_m] and then is the corresponding label/target [y_n]
print(gen[0])
print(gen[1])
>> (array([[6, 7]]), array([8]))
>> (array([[7, 8]]), array([9]))

我想要的是一种简单的方法来提取所有实际可用的目标/标签/基本事实,所以像

print(gen.actual_targets)
>> [8,9]

但我最接近的是

print(gen.targets)
>> [1, 2, 3, 4, 5, 6, 7, 8, 9]

它只给出输入目标,而不是真正使用的目标。那么,我怎样才能从生成器中取出实际可用的目标呢?谢谢

4

1 回答 1

1

也许你可以使用TimeseriesGenerator.start_indexand TimeseriesGenerator.end_index

gen.targets[gen.start_index:gen.end_index + 1]
[8, 9]
于 2021-01-07T15:26:56.930 回答