From 2e86cf1b01ae0ed69f72bf8054330440d432eeb7 Mon Sep 17 00:00:00 2001
From: Yanbo Liang <ybliang8@gmail.com>
Date: Thu, 5 Nov 2015 09:00:03 -0800
Subject: [PATCH] [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.
---
python/pyspark/ml/regression.py | 24 ++++++++++++++++++++++++
1 file changed, 24 insertions(+)
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
--
GitLab