C++计算AUC

刀么克瑟拉莫

于 2023-02-09 18:14:39 发布

阅读量336

点赞数

CC 4.0 BY-SA版权

分类专栏：机器学习 deeplearning 文章标签： c++ 算法

本文链接：https://blog.csdn.net/random_repick/article/details/128957845

该代码实现了一个按预测值和标签升序排序的数组，用于计算AUC（AreaUndertheCurve）。在遍历过程中处理了正负样本，考虑了预测值相同的情况，特别地，当遇到正样本时，会更新累计和sum。最后返回AUC值。

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

按预测值和标签升序排序
遍历数组，遇到负样本neg+1，遇到正样本sum+=neg
考虑到存在正负样本预测值相同，需要加0.5的情况，增加辅助变量tmp和val
理论看这里

#include <iostream>
#include <algorithm>
#include <vector>

using namespace std;

struct val_np {
   
  double val;
  char np;
};

bool auc_cmp(val_np a, val_np b) {
   
  if(a.val == b.val) return a.np < b.np;
  return a.val < b.val;
}

double AUC(vector<val_np> v) {
   
  sort(v.begin(),