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

它應該可以工作。

但不用說您不會獲得任何性能改進。

Answer 2

該錯誤意味着 Numba 無法在nopython（ @jit(nopython=True) ）模式（ @njit是其別名）下 JIT function。

您需要使用常規的@jit ，但加速很可能不會那么劇烈。

同樣代表scipy.stats ，很可能。