import numpy as np
from sklearn import metrics
y = np.array([1, 1, 2, 2])
scores = np.array([0.1, 0.4, 0.35, 0.8])
fpr, tpr, thresholds = metrics.roc_curve(y, scores, pos_label=2)
我正在使用算法进行链接预测,我有一个测试和训练网络。对于给定的节点,我有以下向量[1,0,1,0,0],这意味着该算法可以正确地预测第一个和第三个链接,其他人失败。 现在我想使用scikit学习使用ROC曲线测量算法的性能,在教程中我理解y数组与我的向量相同但是教程中的得分数组是什么?
答案 0 :(得分:0)
根据我的理解,你想使用ROC曲线进行链接预测。这意味着你想测试你的算法进行链接预测,就像ML算法一样进行分类。所以scikit中的得分数组将是由与测试网络相比,训练网络中预测的新链路的链路预测算法。