Question

所以我在numba中有一个由于某种原因未编译的函数（计算ROC曲线下的面积）。我也不确定如何调试该函数，因为在numba装饰函数中设置断点时调试器无法正常工作。

功能如下：

@nb.njit()
def auc_numba(fcst, obs):
    L = obs.size

    i_ord = fcst.argsort()

    sumV = 0.
    sumV2 = 0.
    sumW = 0.
    sumW2 = 0.
    n = 0
    m = 0
    i = 0

    while True:
        nn = mm = 0
        while True:
            j = i_ord[i]
            if obs[j]:
                mm += 1
            else:
                nn += 1
            if i == L - 1:
                break
            jp1 = i_ord[i + 1]
            if fcst[j] != fcst[jp1]:
                break
            i += 1
        sumW += nn * (m + mm / 2.0)
        sumW2 += nn * (m + mm / 2.0) * (m + mm / 2.0)
        sumV += mm * (n + nn / 2.0)
        sumV2 += mm * (n + nn / 2.0) * (n + nn / 2.0)
        n += nn
        m += mm
        i += 1
        if i >= L:
            break

    theta = sumV / (m * n)
    v = sumV2 / ((m - 1) * n * n) - sumV * sumV / (m * (m - 1) * n * n)
    w = sumW2 / ((n - 1) * m * m) - sumW * sumW / (n * (n - 1) * m * m)

    sd_auc = np.sqrt(v / m + w / n)

    return np.array([theta, sd_auc])

我在想的是我实现的while循环有问题。类型很可能是错误的，因此未激活中断并且该功能将永远运行。

以下是一些要测试的示例数据：

obs = np.array([1, 0, 1, 1, 0, 1, 0, 1, 1, 0, 0, 1, 0, 0, 1, 1, 1, 0, 1, 1, 1, 0, 1, 1, 0, 0, 1])
fcst = np.array([0.7083333, 0.5416667, 0.875, 0.5833333, 0.2083333, 0.8333333, 0.1666667, 0.9583333, 0.625, 0.1666667, 0.5, 1.0, 0.6666667, 0.2083333, 0.875, 0.75, 0.625, 0.3333333, 0.8333333, 0.2083333, 0.125, 0.0, 0.875, 0.8333333, 0.125, 0.5416667, 0.75])

当我在没有装饰器的情况下运行此程序时，我得到[0.89488636 0.06561209]是正确的值。

所以我想我是否可以得到一些帮助，以了解为什么它不进行编译，以及有关如何在numba中进行调试的一些技巧？

Answer 1

双while True循环发生了一些奇怪的事情。出于某种原因（我不理解），如果您在顶部创建两个变量x和y，然后：

x = 1
y = 0

while True:
    nn = mm = 0
    while x > y:

并保持其他所有内容不变，代码即可正常工作。我要向Numba跟踪器提交问题，因为这对我来说似乎是个错误。

更新：可以发现here

的数字问题

numba函数将无法编译

1 个答案: