Skip to content
Snippets Groups Projects
Commit 2e86cf1b authored by Yanbo Liang's avatar Yanbo Liang Committed by Xiangrui Meng
Browse files

[SPARK-11527][ML][PYSPARK] PySpark AFTSurvivalRegressionModel should expose...

[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.
parent 72634f27
No related branches found
No related tags found
No related merge requests found
...@@ -824,6 +824,30 @@ class AFTSurvivalRegressionModel(JavaModel): ...@@ -824,6 +824,30 @@ class AFTSurvivalRegressionModel(JavaModel):
.. versionadded:: 1.6.0 .. 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): def predictQuantiles(self, features):
""" """
Predicted Quantiles Predicted Quantiles
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment