如何在遷移學習 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))