Keras 如何使用自定义生成器生成混淆矩阵

我已经实现了我自己的自定义数据生成器，我正在使用

model.predict\u generator（generator=testing\u generator，steps=steps，verbose=0）

生成我的预测。为了生成混淆图，我正在使用：

conf_mat = confusion_matrix(testing_generator.classes, y_predict,labels=classes)

我犯了一个错误

“生成器”对象没有属性“类”

这显然是因为我没有实现任何从生成器返回所有GT的方法。我不知道如何实施这样的事情

我的生成器具有一般形式：

    def generator_interference(self,number_of_steps_per_batch):   

        for i in range(number_of_steps_per_batch):
            """
            Some code to generates batches of samples
            :return: 
            """
            yield features, labels

如何检索GT以创建混淆矩阵？我可以简单地添加一个名为

self.classes

的变量并累加标签吗？如果是这样，我如何在预测之间清除它