This commit is contained in:
Carterla8 2023-05-22 17:39:52 -05:00 committed by GitHub
commit db98fc1e59
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -118,9 +118,8 @@ private[this] class RawAnnoyQueryIndex[D <: Distance[D]](
} }
} }
// Annoy java lib do not expose param for numOfNodesToExplore. // The annoy-java lib does not expose the param for numOfNodesToExplore (the default value is numOfTrees*numOfNeighbours).
// Default number is numOfTrees*numOfNeigbours. // As a workaround, we can artificially increase the numOfNeighbours to be requested and then cap it before returning.
// Simple hack is to artificially increase the numOfNeighbours to be requested and then just cap it before returning.
private[this] def neighboursToRequest( private[this] def neighboursToRequest(
numOfNeighbours: Int, numOfNeighbours: Int,
annoyParams: AnnoyRuntimeParams annoyParams: AnnoyRuntimeParams