Fix the enables larger Datasets (#35)

scverse · Aug 4, 2023 · 404fd70 · 404fd70
1 parent e48472d
commit 404fd70
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rapids_singlecell/scanpy_gpu/_harmonpy_gpu.py b/rapids_singlecell/scanpy_gpu/_harmonpy_gpu.py
@@ -211,7 +211,7 @@ def allocate_buffers(self):
 
     def init_cluster(self):
         # Start with cluster centroids
-        kmeans_obj = KMeans(n_clusters=self.K, init="k-means++").fit(self.Z_cos.T)
+        kmeans_obj = KMeans(n_clusters=self.K, init="k-means||").fit(self.Z_cos.T)
         self.Y = kmeans_obj.cluster_centers_.T
         # (1) Normalize
         self.Y = self.Y / cp.linalg.norm(self.Y, ord=2, axis=0)