lua - Torch/Lua，如何计算曲线下面积(AUC)？

Question

我是 Torch 的新手，我正在研究一个问题，我必须描绘接受者操作特征 (ROC) 曲线及其曲线下面积 (AUC)。

我必须数组:y 轴上的 TPrate 和 x 轴上的 FPrate，两者都具有大小 n

如何在 Torch7 或 Lua 中计算此函数的曲线下面积？

我还找到了this code from JayClascoe.com但我不知道如何使用它的功能。

Answer 1

您将无法使用 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