From 7859590b284637d5ac0bd449b78531b4b0fc168c Mon Sep 17 00:00:00 2001 From: zero323 Date: Mon, 30 Sep 2019 22:57:16 +0200 Subject: [PATCH] Add solver and offsetCol to GeneralizedLinearRegression - Add missing Params - Adds missing arguments - Adds missing getter and setter Resolves #253 --- third_party/3/pyspark/ml/regression.pyi | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/third_party/3/pyspark/ml/regression.pyi b/third_party/3/pyspark/ml/regression.pyi index 993e81e7..80e3bb4f 100644 --- a/third_party/3/pyspark/ml/regression.pyi +++ b/third_party/3/pyspark/ml/regression.pyi @@ -233,8 +233,10 @@ class GeneralizedLinearRegression(JavaEstimator[GeneralizedLinearRegressionModel linkPredictionCol: Param variancePower: Param linkPower: Param - def __init__(self, *, labelCol: str = ..., featuresCol: str = ..., predictionCol: str = ..., family: str = ..., link: Optional[str] = ..., fitIntercept: bool = ..., maxIter: int = ..., tol: float = ..., regParam: float = ..., weightCol: Optional[str] = ..., solver: str = ..., linkPredictionCol: Optional[str] = ..., variancePower: float = ..., linkPower: Optional[float] = ...) -> None: ... - def setParams(self, *, labelCol: str = ..., featuresCol: str = ..., predictionCol: str = ..., family: str = ..., link: Optional[str] = ..., fitIntercept: bool = ..., maxIter: int = ..., tol: float = ..., regParam: float = ..., weightCol: Optional[str] = ..., solver: str = ..., linkPredictionCol: Optional[str] = ..., variancePower: float = ..., linkPower: Optional[float] = ...) -> GeneralizedLinearRegression: ... + solver: Param + offsetCol: Param + def __init__(self, *, labelCol: str = ..., featuresCol: str = ..., predictionCol: str = ..., family: str = ..., link: Optional[str] = ..., fitIntercept: bool = ..., maxIter: int = ..., tol: float = ..., regParam: float = ..., weightCol: Optional[str] = ..., solver: str = ..., linkPredictionCol: Optional[str] = ..., variancePower: float = ..., linkPower: Optional[float] = ..., offsetCol: Optional[str] = ...) -> None: ... + def setParams(self, *, labelCol: str = ..., featuresCol: str = ..., predictionCol: str = ..., family: str = ..., link: Optional[str] = ..., fitIntercept: bool = ..., maxIter: int = ..., tol: float = ..., regParam: float = ..., weightCol: Optional[str] = ..., solver: str = ..., linkPredictionCol: Optional[str] = ..., variancePower: float = ..., linkPower: Optional[float] = ..., offsetCol: Optional[str] = ...) -> GeneralizedLinearRegression: ... def setFamily(self, value: str) -> GeneralizedLinearRegression: ... def getFamily(self) -> str: ... def setLinkPredictionCol(self, value: str) -> GeneralizedLinearRegression: ... @@ -245,6 +247,8 @@ class GeneralizedLinearRegression(JavaEstimator[GeneralizedLinearRegressionModel def getVariancePower(self) -> float: ... def setLinkPower(self, value: float) -> GeneralizedLinearRegression: ... def getLinkPower(self) -> float: ... + def setOffsetCol(self, value: str) -> GeneralizedLinearRegression: ... + def getOffsetCol(self) -> str: ... class GeneralizedLinearRegressionModel(JavaModel, JavaPredictionModel, JavaMLWritable, JavaMLReadable[GeneralizedLinearRegressionModel], HasTrainingSummary): @property