Update RawAnnoyQueryIndex.scala

Simple suggestion to improve comment.
This commit is contained in:
Carterla8 2023-04-01 08:05:00 +01:00 committed by GitHub
parent ec83d01dca
commit 2a31ca9cb1
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