-
- Downloads
[SPARK-19704][ML] AFTSurvivalRegression should support numeric censorCol
## What changes were proposed in this pull request? make `AFTSurvivalRegression` support numeric censorCol ## How was this patch tested? existing tests and added tests Author: Zheng RuiFeng <ruifengz@foxmail.com> Closes #17034 from zhengruifeng/aft_numeric_censor.
Showing
- mllib/src/main/scala/org/apache/spark/ml/regression/AFTSurvivalRegression.scala 3 additions, 3 deletions...rg/apache/spark/ml/regression/AFTSurvivalRegression.scala
- mllib/src/main/scala/org/apache/spark/ml/regression/IsotonicRegression.scala 1 addition, 1 deletion...a/org/apache/spark/ml/regression/IsotonicRegression.scala
- mllib/src/test/scala/org/apache/spark/ml/regression/AFTSurvivalRegressionSuite.scala 33 additions, 1 deletion...ache/spark/ml/regression/AFTSurvivalRegressionSuite.scala
Loading
Please register or sign in to comment