aboutsummaryrefslogtreecommitdiff
path: root/python
diff options
context:
space:
mode:
authorYanbo Liang <ybliang8@gmail.com>2015-11-05 09:00:03 -0800
committerXiangrui Meng <meng@databricks.com>2015-11-05 09:00:03 -0800
commit2e86cf1b01ae0ed69f72bf8054330440d432eeb7 (patch)
treef1d039072d69c8a17b518cb9768fa8b8beea22fc /python
parent72634f27e3110fd7f5bfca498752f69d0b1f873c (diff)
downloadspark-2e86cf1b01ae0ed69f72bf8054330440d432eeb7.tar.gz
spark-2e86cf1b01ae0ed69f72bf8054330440d432eeb7.tar.bz2
spark-2e86cf1b01ae0ed69f72bf8054330440d432eeb7.zip
[SPARK-11527][ML][PYSPARK] PySpark AFTSurvivalRegressionModel should expose coefficients/intercept/scale
PySpark ```AFTSurvivalRegressionModel``` should expose coefficients/intercept/scale. mengxr vectorijk Author: Yanbo Liang <ybliang8@gmail.com> Closes #9492 from yanboliang/spark-11527.
Diffstat (limited to 'python')
-rw-r--r--python/pyspark/ml/regression.py24
1 files changed, 24 insertions, 0 deletions
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