From 2e86cf1b01ae0ed69f72bf8054330440d432eeb7 Mon Sep 17 00:00:00 2001 From: Yanbo Liang Date: Thu, 5 Nov 2015 09:00:03 -0800 Subject: [SPARK-11527][ML][PYSPARK] PySpark AFTSurvivalRegressionModel should expose coefficients/intercept/scale PySpark ```AFTSurvivalRegressionModel``` should expose coefficients/intercept/scale. mengxr vectorijk Author: Yanbo Liang Closes #9492 from yanboliang/spark-11527. --- python/pyspark/ml/regression.py | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) (limited to 'python/pyspark') diff --git a/python/pyspark/ml/regression.py b/python/pyspark/ml/regression.py index ab26616f4a..d7b4fd92c3 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 -- cgit v1.2.3