获取model.predict_generator的测试准确性

时间:2018-08-20 21:34:22

标签: python keras

我正在调用model.predict_generator()方法来对测试数据集进行预测。我的问题是如何获取测试准确性?以下是我的代码行:

predictions = model.predict_generator(test_images, steps=3,  verbose=0)

test_iamges函数的参数predict_generator中具有通过调用ImageDataGenerator从本地磁盘检索的批处理数据

任何指导将不胜感激。

谢谢

1 个答案:

答案 0 :(得分:3)

有一个相应的函数model.evaluate_generator,它会给出测试集上的损失,acc等,但不会给出预测。看一下模型documentation,例如:

loss, acc = model.evaluate_generator(test_images, steps=3, verbose=0)
  

返回:标量测试损失(如果模型具有单个输出且没有度量标准)或标量列表(如果模型具有多个输出和/或度量标准)。