-
- Downloads
[SPARK-8518] [ML] Log-linear models for survival analysis
[Accelerated Failure Time (AFT) model](https://en.wikipedia.org/wiki/Accelerated_failure_time_model) is the most commonly used and easy to parallel method of survival analysis for censored survival data. It is the log-linear model based on the Weibull distribution of the survival time. Users can refer to the R function [```survreg```](https://stat.ethz.ch/R-manual/R-devel/library/survival/html/survreg.html) to compare the model and [```predict```](https://stat.ethz.ch/R-manual/R-devel/library/survival/html/predict.survreg.html) to compare the prediction. There are different kinds of model prediction, I have just select the type ```response``` which is default used for R. Author: Yanbo Liang <ybliang8@gmail.com> Closes #8611 from yanboliang/spark-8518.
Showing
- mllib/src/main/scala/org/apache/spark/ml/regression/AFTSurvivalRegression.scala 449 additions, 0 deletions...rg/apache/spark/ml/regression/AFTSurvivalRegression.scala
- mllib/src/test/scala/org/apache/spark/ml/regression/AFTSurvivalRegressionSuite.scala 311 additions, 0 deletions...ache/spark/ml/regression/AFTSurvivalRegressionSuite.scala
Loading
Please register or sign in to comment