Commit a8f1170b authored by Julien Jerphanion's avatar Julien Jerphanion

Remove old script

parent 32e855bc
import numpy as np
import pytest
import kdtree
from sklearn.neighbors import KDTree
if __name__ == '__main__':
n = 1000
n_query = 100
d = 10
k = 10
np.random.seed(1)
X = np.random.rand(n, d)
query_points = np.random.rand(n_query, d)
tree = kdtree.KDTree(X, leaf_size=256)
closests = np.zeros((n_query, k), dtype=np.int32)
# tree.query(query_points, closests)
# skl_tree = KDTree(X, leaf_size=256)
# skl_closests = skl_tree.query(query_points, k=k, return_distance=False).astype(np.int32)
\ No newline at end of file
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment