我已经训练/微调了几个Keras模型,在此期间,我使用“准确性”作为唯一的指标来计算。现在,在训练了很长一段时间后,我意识到我需要准确性和回忆率。在训练完成后,我是否有机会使用保存的模型和手头的准确性提取/计算该信息?
roqulrg31#
您可以对测试集应用已训练的模型,生成真标签和预测标签的列表,然后使用scikit-learn中的得分度量。
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)
1条答案
按热度按时间roqulrg31#
您可以对测试集应用已训练的模型,生成真标签和预测标签的列表,然后使用
scikit-learn
中的得分度量。例如,对于二元分类问题: