计算查准率、查全率代码
时间: 2023-12-12 10:03:17 浏览: 61
以下是计算查准率、查全率的示例代码:
```python
# 实际结果
actual = [1, 0, 1, 1, 0, 1, 0, 0, 1, 1]
# 预测结果
predicted = [1, 0, 1, 0, 1, 0, 1, 0, 1, 0]
# 计算真正例数、假正例数、真反例数、假反例数
tp = fp = tn = fn = 0
for i in range(len(actual)):
if actual[i] == 1 and predicted[i] == 1:
tp += 1
elif actual[i] == 0 and predicted[i] == 1:
fp += 1
elif actual[i] == 0 and predicted[i] == 0:
tn += 1
elif actual[i] == 1 and predicted[i] == 0:
fn += 1
# 计算查准率、查全率
precision = tp / (tp + fp)
recall = tp / (tp + fn)
print("查准率:", precision)
print("查全率:", recall)
```
输出:
```
查准率: 0.5714285714285714
查全率: 0.6666666666666666
```
以上代码中,实际结果和预测结果分别表示真实标签和预测标签,通过循环遍历实际结果和预测结果,计算真正例数(tp)、假正例数(fp)、真反例数(tn)、假反例数(fn),然后根据公式计算查准率(precision)和查全率(recall)。
其中,查准率表示预测为正例的样本中,实际为正例的比例;查全率表示实际为正例的样本中,被正确预测为正例的比例。