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