什么是scikit中ROC曲线中的预测数组

时间:2017-11-30 15:44:26

标签: machine-learning scikit-learn statistics roc

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数组与我的向量相同但是教程中的得分数组是什么?

1 个答案:

答案 0 :(得分:0)

根据我的理解,你想使用ROC曲线进行链接预测。这意味着你想测试你的算法进行链接预测,就像ML算法一样进行分类。所以scikit中的得分数组将是由与测试网络相比,训练网络中预测的新链路的链路预测算法。