每当我尝试在我的类方法上使用 @jit 时，我都会收到 IndentationError: unexpected indent

Question

我几天来一直在努力让@jit 工作以加快我的代码速度。 最后我遇到了这个，描述了在对象方法中添加@jit： http://williamjshipman.wordpress.com/2013/12/24/learning-python-eight-ways-to-filter-an-image

我有一个名为GentleBoostC 的类，我想加快其中名为train 的方法。 train 接受三个参数（一个二维数组、一个一维数组和一个整数），并且不返回任何内容。

这是我在代码中的内容：

import numba
from numba import jit, autojit, int_, void, float_, object_


class GentleBoostC(object):
    # lots of functions

    # and now the function I want to speed up
    @jit (void(object_,float_[:,:],int_[:],int_)) 
    def train(self, X, y, H):
        # do stuff

但我不断收到缩进错误，指向定义 train 函数的行。我的缩进没有任何问题。我重新缩进了我的整个代码。如果我用@jit 注释掉这一行，那么就没有问题了。

这是确切的错误：

   @jit (void(object_,float_[:,:],int_[:],int_))
  File "C:\Users\app\Anaconda\lib\site-packages\numba\decorators.py", line 224, in _jit_decorator
    nopython=nopython, func_ast=func_ast, **kwargs)
  File "C:\Users\app\Anaconda\lib\site-packages\numba\decorators.py", line 133, in compile_function
    func_env = pipeline.compile2(env, func, restype, argtypes, func_ast=func_ast, **kwds)
  File "C:\Users\app\Anaconda\lib\site-packages\numba\pipeline.py", line 133, in compile2
    func_ast = functions._get_ast(func)
  File "C:\Users\app\Anaconda\lib\site-packages\numba\functions.py", line 89, in _get_ast
    ast.PyCF_ONLY_AST | flags, True)
  File "C:\Users\app\Documents\Python Scripts\gentleboost_c_class_jit_v5_nolimit.py", line 1
    def train(self, X, y, H):
    ^
IndentationError: unexpected indent

Answer 1

根据我从文档中看到的，您不能将装饰器应用于方法；您看到的错误来自 JIT 解析器在不在 class 语句的上下文中时未处理源代码缩进。

如果您希望编译该方法的主体，则需要将其分解为一个单独的函数，并从该方法中调用该函数：

@jit(void(object_, float_[:,:], int_[:], int_)) 
def train_function(instance, X, y, H):
    # do stuff

class GentleBoostC(object):
    def train(self, X, y, H):
        train_function(self, X, y, H)