GMM blockify input vectors (#482)

Resolves #426
zero323 · Aug 29, 2020 · 48381b0 · 48381b0
1 parent 2bee703
commit 48381b0
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/third_party/3/pyspark/ml/clustering.pyi b/third_party/3/pyspark/ml/clustering.pyi
@@ -55,6 +55,7 @@ class _GaussianMixtureParams(
     HasTol,
     HasAggregationDepth,
     HasWeightCol,
+    HasBlockSize,
 ):
     k: Param[int]
     def getK(self) -> int: ...
@@ -97,7 +98,8 @@ class GaussianMixture(
         maxIter: int = ...,
         seed: Optional[int] = ...,
         aggregationDepth: int = ...,
-        weightCol: Optional[str] = ...
+        weightCol: Optional[str] = ...,
+        blockSize: int = ...
     ) -> None: ...
     def setParams(
         self,
@@ -110,7 +112,8 @@ class GaussianMixture(
         maxIter: int = ...,
         seed: Optional[int] = ...,
         aggregationDepth: int = ...,
-        weightCol: Optional[str] = ...
+        weightCol: Optional[str] = ...,
+        blockSize: int = ...
     ) -> GaussianMixture: ...
     def setK(self, value: int) -> GaussianMixture: ...
     def setMaxIter(self, value: int) -> GaussianMixture: ...
@@ -121,6 +124,7 @@ class GaussianMixture(
     def setSeed(self, value: int) -> GaussianMixture: ...
     def setTol(self, value: float) -> GaussianMixture: ...
     def setAggregationDepth(self, value: int) -> GaussianMixture: ...
+    def setBlockSize(self, value: int) -> GaussianMixture: ...
 
 class GaussianMixtureSummary(ClusteringSummary):
     @property