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)