impute.py 文件源码

def _make_kdtree(self, x):
        self.kdtree = cKDTree(mpiops.random_full_points(x, Napprox=self.nodes))
        if not np.isfinite(self.kdtree.query(x, k=self.k)[0]).all():
            log.warning('Kdtree computation encountered problem. '
                        'Not enough neighbors available to compute '
                        'kdtree. Printing kdtree for debugging purpose')
            raise ValueError('Computed kdtree is not fully populated.'
                             'Not enough valid neighbours available.')