diff options
Diffstat (limited to 'lib/mining/knn/knn.go')
| -rw-r--r-- | lib/mining/knn/knn.go | 5 |
1 files changed, 1 insertions, 4 deletions
diff --git a/lib/mining/knn/knn.go b/lib/mining/knn/knn.go index 977e6dc6..71c60cc9 100644 --- a/lib/mining/knn/knn.go +++ b/lib/mining/knn/knn.go @@ -80,10 +80,7 @@ func (in *Runtime) FindNeighbors(samples *tabula.Rows, instance *tabula.Row) ( } // Make sure number of neighbors is greater than request. - minK := in.AllNeighbors.Len() - if minK > in.K { - minK = in.K - } + minK := min(in.AllNeighbors.Len(), in.K) kneighbors = in.AllNeighbors.SelectRange(0, minK) |