use l2expanded distance in kmeans transform

rapidsai · Oct 2, 2024 · 176c9a9 · 176c9a9
1 parent 9f805cd
commit 176c9a9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/cuml/cuml/cluster/kmeans.pyx b/python/cuml/cuml/cluster/kmeans.pyx
@@ -604,7 +604,7 @@ class KMeans(UniversalBase,
             cdef KMeansParams* params = \
                 <KMeansParams*><size_t>self._get_kmeans_params()
 
-            params.metric = CuvsDistanceType.L2SqrtExpanded
+            params.metric = CuvsDistanceType.L2Expanded
 
             int_dtype = np.int32 if self.labels_.dtype == np.int32 else np.int64