python - 获得 model.predict_generator 的测试精度

Question

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

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

其中predict_generator函数参数中的test_iamges有通过调用ImageDataGenerator从本地磁盘获取的批量数据

任何指导将不胜感激。

谢谢

Answer 1

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

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

Returns: Scalar test loss (if the model has a single output and no metrics) or list of scalars (if the model has multiple outputs and/or metrics).