如何在迁移学习 vgg16 模型中获得准确率、召回率、f1 分数？

Question

如何获得精度、召回率、混淆矩阵和 f1 分数？ 我是深度学习的新手。 任何人都可以给我资源或帮助我查找信息。

Answer 1

从sklearn导入模块：

from sklearn.metrics import classification_report, confusion_matrix

假设你的测试集的解释变量包含在X_test ，在响应变量y_test ，和你的模型被命名为model ，预测了测试设置的值：

y_pred = model.predict(X_test)

然后打印分类报告和混淆矩阵：

print(classification_report(y_test, y_pred))
print(confusion_matrix(y_test, y_pred))