From 0c1053045cbb24dad7b416e7eb3db4523a492c16 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Luis=20Eduardo=20Ram=C3=ADrez=20Montoya?=
 <luiseduardoramirezmontoya@ciencias.unam.mx>
Date: Tue, 2 Jan 2024 18:30:35 -0600
Subject: [PATCH] Add init to kmeans

---
 src/orbit_determination/outlier_rejection.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/orbit_determination/outlier_rejection.jl b/src/orbit_determination/outlier_rejection.jl
index 403c46b2..f81144d6 100644
--- a/src/orbit_determination/outlier_rejection.jl
+++ b/src/orbit_determination/outlier_rejection.jl
@@ -160,7 +160,7 @@ function outlier_rejection(radec::Vector{RadecMPC{T}}, sol::NEOSolution{T, T},
             points[2, i] = N_outliers[i]
         end
         # K-means clustering
-        cluster = kmeans(points, 2)
+        cluster = kmeans(points, 2; init = [1, max_drop + 1])
         # Index of smallest cluster
         i_0 = cluster.assignments[1]
         # Find last fit of smallest cluster