链路预测precision实现代码
时间: 2023-08-12 14:08:59 浏览: 54
MapReduce环境下的并行复杂网络链路预测——论文
链路预测的 precision 是用来度量预测结果中实际连边占预测结果中连边总数的比例,其计算公式为:
```
precision = TP / (TP + FP)
```
其中,TP 表示 true positive,即预测为连边且实际为连边的数量;FP 表示 false positive,即预测为连边但实际不是连边的数量。
以下是一个简单的链路预测 precision 实现代码,假设预测结果为列表 pred_edges,实际连边为列表 true_edges:
```python
def link_prediction_precision(pred_edges, true_edges):
tp = 0
fp = 0
for edge in pred_edges:
if edge in true_edges:
tp += 1
else:
fp += 1
if tp + fp == 0:
return 0.0
else:
return tp / (tp + fp)
```
需要注意的是,这个实现代码并不包括其它常见的性能度量,如 recall、F1-score 等。在实际使用中,需要结合具体应用场景选择合适的性能度量进行综合评价。
阅读全文