从keras中的TimeseriesGenerator获取实际使用的目标,而不是输入的目标



假设我用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]

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

也许您可以使用TimeseriesGenerator.start_indexTimeseriesGenerator.end_index:

gen.targets[gen.start_index:gen.end_index + 1]
[8, 9]

相关内容

  • 没有找到相关文章

最新更新