我按照 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 开始时运行?有没有更好的方法来实现这一点?