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

Question

我已经实现了我自己的自定义数据生成器，并且我正在使用model.predict_generator(generator = testing_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并累积标签的变量吗？如果是这样，我如何在预测之间清除它？

标签： kerasscikit-learngenerator