链路预测里面的AUC和准确率代码怎么写

链路预测中经常使用的评估指标有AUC和准确率，下面是一个简单的Python代码示例，演示如何使用scikit-learn库来计算这些指标：

import numpy as np
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score, roc_auc_score
from sklearn.ensemble import RandomForestClassifier

# 读取数据
data = np.loadtxt('network_data.csv', delimiter=',')

# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(data[:, :-1], data[:, -1], test_size=0.2)

# 训练模型
clf = RandomForestClassifier()
clf.fit(X_train, y_train)

# 预测
y_pred = clf.predict(X_test)

# 计算准确率
accuracy = accuracy_score(y_test, y_pred)
print('准确率：', accuracy)

# 计算AUC
y_pred_prob = clf.predict_proba(X_test)[:, 1]
auc = roc_auc_score(y_test, y_pred_prob)
print('AUC：', auc)

代码首先读取数据，然后使用`train_test_split`函数将数据划分为训练集和测试集，接着使用随机森林算法进行训练和预测。计算准确率的方法是使用`accuracy_score`函数，将真实标签和预测标签作为参数传入即可。计算AUC的方法是使用`predict_proba`方法得到预测概率，然后使用`roc_auc_score`函数计算AUC值。需要注意的是，在实际应用中还需要进行交叉验证等步骤来评估模型性能。