package/gonum
1
0
Fork 0
mirror of https://github.com/gonum/gonum.git synced 2025-10-09 17:10:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

stat: fix ROC for uniform cases

Browse Source
This commit is contained in:
kortschak
2018-01-30 02:22:43 +10:30
committed by Dan Kortschak
parent 287df04a63
commit 870bcdceb6
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
stat/roc.go
View File

@@ -108,8 +108,13 @@ func ROC(n int, y []float64, classes []bool, weights []float64) (tpr, fpr []floa
fpr = fpr[:(bin + 1)] fpr = fpr[:(bin + 1)]
} }
invNeg := 1 / nNeg var invNeg, invPos float64
invPos := 1 / nPos if nNeg != 0 {
invNeg = 1 / nNeg
}
if nPos != 0 {
invPos = 1 / nPos
}
for i := range tpr { for i := range tpr {
tpr[i] *= invPos tpr[i] *= invPos
fpr[i] *= invNeg fpr[i] *= invNeg