compute_raw_knn.py

%%time 
a = nearest_ngbr_raw(arr)

display_lprof.py

print(open('sim_result', 'r').read())

knn_raw_py.py

%%writefile simulation.py
import numpy as np
def comp_inner_raw(i, x):
    res = np.zeros(x.shape[0], dtype=np.float64)
    for j in range(x.shape[0]):
        res[j] = np.sqrt(np.sum((i-x[j])**2))
    return res
  
def nearest_ngbr_raw(x):
    dist = {}
    for idx,i in enumerate(x):
        lst = comp_inner_raw(i,x)
        s = np.argsort(lst)
        sorted_array = np.array(x)[s][1:]
        dist[idx] = s[1:]
    return dist

load_lprofiler.py

import numpy as np
%load_ext line_profiler

random_array.py

arr = np.random.rand(1000,800)

sim_knn.py

from simulation import nearest_ngbr_raw
%lprun -T sim_result -f nearest_ngbr_raw nearest_ngbr_raw(arr)