e3bb2410a9667cdc224bc1526211f306cd1ae658,pynndescent/pynndescent_.py,NNDescent,prepare,#NNDescent#,1426

Before Change

    def prepare(self):
        self._init_search_graph()
        self._init_search_function()
        return

    def query(self, query_data, k=10, epsilon=0.1):
        Query the training graph_data for the k nearest neighbors

After Change

        return

    def prepare(self):
        if not hasattr(self, "_search_graph"):
            self._init_search_graph()
        if not hasattr(self, "_search_function"):
            self._init_search_function()
            // Force compilation of the search function (hardcoded k, epsilon)
            query_data = self._raw_data[:1]
            _ = self._search_function(
                query_data, 5, 0.0, self._visited, self.search_rng_state.copy(),
            )
        return

    def query(self, query_data, k=10, epsilon=0.1):
        Query the training graph_data for the k nearest neighbors

Instances