Python 中的 Librosa 和 Numba

Question

我正在体验 numba 以加快我的代码速度，我在此代码中遇到错误

@njit
def fee(x):
    chroma_stft = librosa.feature.chroma_stft(y=data, sr=sampling_rate).ravel()
    return(chroma_stft)

TypingError: Failed in nopython mode pipeline (step: nopython frontend) Unknown attribute 'chroma_stft' of type Module(<module 'librosa.feature' from '/usr/local/lib/python3.7/dist-packages/librosa/feature/__init__.py'>)

如何解决这个问题，有没有办法在 numba 中使用 scipy.stats 函数？

谢谢

Answer 1

@njit

是简写

@jit(nopython=True)

不允许导入 Python 代码。改成

@jit

它应该可以工作。

但不用说you will not gain any performance improvement。

Answer 2

该错误意味着 Numba 无法在 nopython (@jit(nopython=True)) 模式下 JIT 函数（@njit 是其别名）。

您需要使用常规的@jit，但加速可能不会那么剧烈。

scipy.stats 最有可能也是这样。