diff --git a/python/pyspark/ml/regression.py b/python/pyspark/ml/regression.py index ab26616f4a01d9daf24f1b1c6d5764f47db7c6a1..d7b4fd92c3817e32c090a59d04ac0deeacddcbb6 100644 --- a/python/pyspark/ml/regression.py +++ b/python/pyspark/ml/regression.py @@ -824,6 +824,30 @@ class AFTSurvivalRegressionModel(JavaModel): .. versionadded:: 1.6.0 """ + @property + @since("1.6.0") + def coefficients(self): + """ + Model coefficients. + """ + return self._call_java("coefficients") + + @property + @since("1.6.0") + def intercept(self): + """ + Model intercept. + """ + return self._call_java("intercept") + + @property + @since("1.6.0") + def scale(self): + """ + Model scale paramter. + """ + return self._call_java("scale") + def predictQuantiles(self, features): """ Predicted Quantiles