word_vector_box.py 文件源码

def index(self, metric='cosine'):
        alg = 'brute' if (metric == 'cosine') else 'auto'
        if not SKLEARN:
            raise WordVectorBoxException("Needs sklearn to work")
        self._nn = NearestNeighbors(metric=metric, algorithm=alg)
        self._nn.fit(self.W)
        return self