2022-05-10AUC_CPP两种方法实现

最新推荐文章于 2025-06-07 00:30:00 发布

lucifer80861

最新推荐文章于 2025-06-07 00:30:00 发布

阅读量259

点赞数

CC 4.0 BY-SA版权

分类专栏： NLP C++ 文章标签： c++ 开发语言

本文链接：https://blog.youkuaiyun.com/lucifer80861/article/details/124658423

C++ 同时被 2 个专栏收录

21 篇文章

订阅专栏

NLP

1 篇文章

订阅专栏

该博客展示了如何计算ROC曲线下的面积（AUC）的两种方法。第一种方法通过分别遍历正例和负例，比较它们的预测概率来计算。第二种方法则是将样本排序后，根据正负例的比例来估算AUC。两种方法都涉及到对预测概率的处理和AUC的累加计算。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

https://zhuanlan.zhihu.com/p/462734871

int main() {
    vector<int> label = { 1,0,0 };
    vector<double> pre = { 0.9, 0.8, 0.3 };
    vector<double> pos;
    vector<double> neg;
    int n = label.size();
    double auc = 0;
    int num = 0;
    for (int i = 0; i < n; ++i) {
        if (label[i] == 0) neg.push_back(pre[i]);
        else pos.push_back(pre[i]);
    }
    for (auto i : pos) {
        for (auto j : neg) {
            num++;
            if (i > j) auc += 1;
            else if (i == j) auc += 0.5;
        }
    }
    cout << auc<<endl;
    auc /= num;
    cout << auc << endl;
    return 0;
}

bool f(pair<int, double> p1, pair<int, double> p2) {
    if (p1.second < p2.second) return true;
    return false;
}

int main() {
    vector<int> label = { 1,0,0 };
    vector<double> pre = { 0.9, 0.8, 0.3 };
    vector<pair<int, double>> v;
    int n = label.size();
    double auc = 0;
    for (int i = 0; i < n; ++i) {
        v.push_back({ label[i], pre[i] });
    }
   // for (int i = 0; i < n;++i) {
    //    cout << v[i].first << " " << v[i].second << endl;
    //}
    sort(v.begin(), v.end(), f);
    //for (int i = 0; i < n; ++i) {
    //    cout << v[i].first << " " << v[i].second << endl;
    //}
    int pos = 0, neg = 0;
    for (int i = 0; i < n; ++i) {
        if (v[i].first == 1) {
            pos++;
            auc = auc + i + 1;
        }
        else neg++;
    }
    auc = (auc - pos * (pos + 1) / 2) / (pos * neg);
    cout << auc;
}

缺一个分数相同加0.5