我正在调用model.predict_generator()
方法来对测试数据集进行预测。我的问题是如何获取测试准确性?以下是我的代码行:
predictions = model.predict_generator(test_images, steps=3, verbose=0)
test_iamges
函数的参数predict_generator
中具有通过调用ImageDataGenerator
从本地磁盘检索的批处理数据
任何指导将不胜感激。
谢谢
答案 0 :(得分:3)
有一个相应的函数model.evaluate_generator
,它会给出测试集上的损失,acc等,但不会给出预测。看一下模型documentation,例如:
loss, acc = model.evaluate_generator(test_images, steps=3, verbose=0)
返回:标量测试损失(如果模型具有单个输出且没有度量标准)或标量列表(如果模型具有多个输出和/或度量标准)。