rmontanana/Platform
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Return AUC 0.5 if nPos==0 || nNeg==0

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2025-05-14 13:15:33 +02:00
parent b639a2d79a
commit e64e281b63
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/main/RocAuc.cpp
View File

@@ -48,6 +48,7 @@ namespace platform {
double tp = 0, fp = 0; double tp = 0, fp = 0;
double totalPos = std::count(y_test.begin(), y_test.end(), classIdx); double totalPos = std::count(y_test.begin(), y_test.end(), classIdx);
double totalNeg = nSamples - totalPos; double totalNeg = nSamples - totalPos;
if (totalPos == 0 || totalNeg == 0) return 0.5; // neutral AUC
for (const auto& [score, label] : scoresAndLabels) { for (const auto& [score, label] : scoresAndLabels) {
if (label == 1) { if (label == 1) {