Clip max neighbors if n < k

rmcaixeta · Apr 25, 2021 · d64aede · d64aede
1 parent 3917271
commit d64aede
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/data_handling.jl b/src/data_handling.jl
@@ -45,7 +45,8 @@ function get_neighbors(origin::AbstractMatrix, target::AbstractMatrix,
 	@assert nhood in [:knn,:radius] "invalid neighborhood type"
 	tree = KDTree(origin)
 	if nhood==:knn
-		idxs, dists = knn(tree, target, neigh_val, true)
+		nneigh = minimum([neigh_val, size(origin,2)])
+		idxs, dists = knn(tree, target, nneigh, true)
 		idxs, dists
 	elseif nhood==:radius
 		idxs = inrange(tree, target, neigh_val)