8 lines
216 B
Python
8 lines
216 B
Python
import numpy as np
|
|
from scipy.spatial import KDTree
|
|
|
|
class KdTreeModel(object):
|
|
def __init__(self,data) -> None:
|
|
self.kdtree_=KDTree(data)
|
|
def query(self,data):
|
|
return self.kdtree_.query(data) |