时间戳

首页 > 解决方案 > Keras 评估生成器

python - Keras 评估生成器

问题描述

我正在尝试使用自定义生成器（用于训练的同一生成器）评估 keras 分类器。预测函数工作得很好，但评估函数抛出 ValueError: Shapes (None, None, None) and (100, 4, 1, 200) is incompatible

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

什么可能导致这种情况？它是同一个生成器，它以相同的形状吐出相同的数据。也尝试使用评估生成器，但同样的错误。

标签： pythonkeras

解决方案

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

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

推荐阅读