Add leafCol to RF RandomForestRegressor (#530)

zero323 · Sep 15, 2020 · f7fa663 · f7fa663
1 parent 98d8d7d
commit f7fa663
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/third_party/3/pyspark/ml/regression.pyi b/third_party/3/pyspark/ml/regression.pyi
@@ -372,6 +372,7 @@ class RandomForestRegressor(
  seed: Optional[int] = ...,
  numTrees: int = ...,
  featureSubsetStrategy: str = ...,
+ leafCol: str = ...,
  minWeightFractionPerNode: float = ...,
  weightCol: Optional[str] = ...,
  bootstrap: Optional[bool] = ...
@@ -394,6 +395,7 @@ class RandomForestRegressor(
  seed: Optional[int] = ...,
  numTrees: int = ...,
  featureSubsetStrategy: str = ...,
+ leafCol: str = ...,
  minWeightFractionPerNode: float = ...,
  weightCol: Optional[str] = ...,
  bootstrap: Optional[bool] = ...