keras 训练模型后计算精度和召回率

kb5ga3dv  于 2023-01-09  发布在  其他
关注(0)|答案(1)|浏览(210)

我已经训练/微调了几个Keras模型,在此期间,我使用“准确性”作为唯一的指标来计算。现在,在训练了很长一段时间后,我意识到我需要准确性和回忆率。在训练完成后,我是否有机会使用保存的模型和手头的准确性提取/计算该信息?

roqulrg3

roqulrg31#

您可以对测试集应用已训练的模型,生成真标签和预测标签的列表,然后使用scikit-learn中的得分度量。

例如,对于二元分类问题:

from sklearn.metrics import precision_score

# I assume y_true are the list of true labels of the test set
# I assume y_pred are the predictions of the model
precision = precision_score(y_true, y_pred)
recall = recall_score(y_true, y_pred)

相关问题