小编Lae*_*ven的帖子

Numba - nopython 模式是否支持元组列表?

我想澄清一下,这是我第一次使用 Numba,所以我距离成为专家还很远。我正在尝试手动实现一个简单的 KNN,代码如下:

@jit(nopython=True)
def knn(training_set, test_set):
for q in range(len(test_set)):
    indexes = [-1]
    values = [np.inf]
    thres = values[-1]

    for u in range(len(training_set)):
        dist = 0
        flag = False
        dist = knn_dist(training_set[u], test_set[q], thres)
        if dist == 0:
            flag = True
        if not flag:

            '''
            Binary search to obtain the index
            '''    

            # Various code

return
Run Code Online (Sandbox Code Playgroud)

现在,我想使用numba的nopython模式来优化代码,以下是部分错误:

---------------------------------------------------------------------------
NotImplementedError                       Traceback (most recent call last)
 in _call_incref_decref(self, builder, root_type, typ, value, funcname, getters)
    185             try:
--> 186                 meminfo = data_model.get_nrt_meminfo(builder, …
Run Code Online (Sandbox Code Playgroud)

python arrays optimization numpy numba

5
推荐指数
1
解决办法
3614
查看次数

标签 统计

arrays ×1

numba ×1

numpy ×1

optimization ×1

python ×1