自定义数据生成器类

时间:2021-06-13 08:14:41

标签: python keras generator

我按照 this tutorial 创建了自己的自定义数据生成器。由于这是一个测试生成器,我希望通过 generator.true_labels 获得单个 epoch 的真实标签。我正在考虑创建一个列表 self.true_labels 并添加行代码:

self.true_labels.extend(keras.utils.to_categorical(y, num_classes=self.n_classes))

__data_generation() 的末尾。问题是,我不确定如何在每个时代之后重置列表。我无法将其放入 on_epoch_end() 中,因为我将无法阅读该列表。是否有类似的函数在每个 epoch 开始时运行?有没有更好的方法来实现这一点?

0 个答案:

没有答案