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

问题描述

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

解决方法

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