我是 Torch 的新手,我正在研究一个问题,我必须描绘接受者操作特征 (ROC) 曲线及其曲线下面积 (AUC)。
我必须数组:y 轴上的 TPrate
和 x 轴上的 FPrate
,两者都具有大小 n
如何在 Torch7 或 Lua 中计算此函数的曲线下面积?
我还找到了this code from JayClascoe.com但我不知道如何使用它的功能。
最佳答案
您将无法使用 JayClascoe 的代码,因为它与函数一起使用,而您的是一组点。假设 FPrate
具有 x 坐标排序的 x 值并且 TPrate
具有相应的值,您可以使用 Trapezoidal rule .这样的事情可能会起作用:
local area = 0
for i = 2, n do
area = area + (FPrate[i] - FPrate[i-1]) * (TPrate[i-1] + TPrate[i])/2
end
关于lua - Torch/Lua,如何计算曲线下面积(AUC)?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/21869157/