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

【问题标题】：what is the prediction array in ROC curves in scikitscikit中ROC曲线中的预测数组是什么
【发布时间】：2018-05-14 14:33:33
【问题描述】：

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 learn 使用 ROC 曲线来测量算法的性能，在教程中我理解了与我的向量相同的 y 数组，但是教程中的分数数组是什么？

【问题讨论】：

假设这是一个简单的二元问题，输出为 1 和 0。分数将来自 scikit-learn 估计器的正类 (1) 的预测概率。在您尝试使用的估算器中查找predict_proba() 或decision_function()。如需更多了解，请在您关于数据、分类器的问题中添加必要的详细信息。

标签： machine-learning scikit-learn statistics roc

【解决方案1】：

据我了解，您想使用 ROC 曲线进行链接预测。这意味着您想测试链接预测的算法，就像用于分类的 ML 算法一样。所以 scikit learn 中的分数数组将是与测试网络相比，训练网络中预测的新链接的链接预测算法。

【讨论】：