python - 如何在 python 中找到已经训练和保存的模型的 f1 分数

Question

假设我有一个已经训练和保存的模型（使用 python）。现在，如果我想获得该模型的 f1 分数，那么如何使用 python 来做到这一点？任何知道这一点的人请帮忙。

标签： pythonkerasmodel

Answer 1

y_true 是您已经拥有并测试您的模型的结果列表。

y_pred 是您的模型预测的列表。

from sklearn.metrics import f1_score
y_true = [0, 1, 2, 0, 1, 2]
y_pred = [0, 2, 1, 0, 0, 1]

f1_score(y_true, y_pred, average='macro')
0.26
f1_score(y_true, y_pred, average='micro')
0.33
f1_score(y_true, y_pred, average='weighted')
0.26
f1_score(y_true, y_pred, average=None)
array([0.8, 0. , 0. ])

如果有效，您可以将其标记为已解决。