Fix init handling of queue_size
@@ -7,13 +7,14 @@ class PyNNDescent(BaseANN):
self._n_neighbors = int(n_neighbors)
self._n_trees = int(n_trees)
self._leaf_size = int(leaf_size)
+ self._queue_size=None
self._pynnd_metric = {'angular': 'cosine',
'euclidean': 'euclidean',
'hamming': 'hamming',
'jaccard': 'jaccard'}[metric]
self.name = 'PyNNDescent(n_neighbors=%d,n_trees=%d,leaf_size=%d,queue_size=%.2f)' % \
- (self._n_neighbors, self._n_trees, self._leaf_size, self._queue_size)
+ (self._n_neighbors, self._n_trees, self._leaf_size)
def fit(self, X):
self._index = pynndescent.NNDescent(X,
