在Keras中定义自定义生成器的属性

时间:2019-04-25 07:26:47

标签: python keras generator

我创建了一个自定义生成器,以便从两个不同的网络按顺序生成数据,然后再将其提供给LSTM网络。

问题是,当我尝试使用通常为默认生成器定义的某些属性(即:samples,batch_size,classs)时,会引发类似以下的错误

AttributeError: 'generator' object has no attribute 'samples'

这确实有意义,因为它不是为自定义对象定义的。我的问题是,如何为自定义属性定义这些属性?我可以通过直接计算和写入前两个属性的值来牺牲灵活性,但是对于类,我看不到如何手动完成。

这是发生问题的行:

cm = confusion_matrix(testgenerator.classes, predictions)

在产生samples = Label的位置之前,在自定义生成器中添加一行代码是个好主意吗?还是有一种更清洁的更合适的方法?

谢谢!!

0 个答案:

没有答案