Keras评估生成器

时间:2020-07-23 17:17:16

标签: python keras

我正在尝试使用自定义生成器(用于训练的同一生成器)评估keras分类器。预测函数可以正常工作,但是评估函数会引发ValueError:形状(无,无,无)和(100、4、1、200)形状不兼容

output = classifier.predict(x=generator)
evaluate = classifier.evaluate(x=generator)

这可能导致什么?这是同一个生成器,它以相同的形状吐出相同的数据。也尝试过使用validate_generator进行尝试,但存在相同的错误。

1 个答案:

答案 0 :(得分:0)

弄清楚了。这是因为我添加了另一个输出。

classifier.summary()
classifier = Model(inputs=classifier.input,
               outputs=[classifier.output, classifier.get_layer('attention').output])
classifier.compile(loss='categorical_crossentropy', optimizer=SGD(), metrics=['accuracy'])
相关问题