diff options
author | Yanbo Liang <ybliang8@gmail.com> | 2015-11-05 09:00:03 -0800 |
---|---|---|
committer | Xiangrui Meng <meng@databricks.com> | 2015-11-05 09:00:03 -0800 |
commit | 2e86cf1b01ae0ed69f72bf8054330440d432eeb7 (patch) | |
tree | f1d039072d69c8a17b518cb9768fa8b8beea22fc | |
parent | 72634f27e3110fd7f5bfca498752f69d0b1f873c (diff) | |
download | spark-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.
-rw-r--r-- | python/pyspark/ml/regression.py | 24 |
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 |