如何在迁移学习vgg16模型中获得精度，召回率，f1分数？

Question

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

Answer 1

从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))

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

df5 = df3.withColumnRenamed("sum(count)","destination_total")