[Case Study]AFT Survival Regression Analysis using Scala

是 否 +2 论坛币

k人 参与回答

经管之家送您一份

应届毕业生专属福利!

求职就业群

赵安豆老师微信：zhaoandou666

经管之家联合CDA

送您一个全额奖学金名额~ !

立即领取

感谢您参与论坛问题回答

经管之家送您两个论坛币！

+2 论坛币

1. // scalastyle:off println
   
2. package org.apache.spark.examples.ml
   
3. 
   
4. import org.apache.spark.sql.SQLContext
   
5. import org.apache.spark.{SparkContext, SparkConf}
   
6. // $example on$
   
7. import org.apache.spark.ml.regression.AFTSurvivalRegression
   
8. import org.apache.spark.mllib.linalg.Vectors
   
9. // $example off$
   
10. 
    
11. /**
    
12. * An example for AFTSurvivalRegression.
    
13. */
    
14. object AFTSurvivalRegressionExample {
    
15. 
    
16.   def main(args: Array[String]): Unit = {
    
17.     val conf = new SparkConf().setAppName("AFTSurvivalRegressionExample")
    
18.     val sc = new SparkContext(conf)
    
19.     val sqlContext = new SQLContext(sc)
    
20. 
    
21.     // $example on$
    
22.     val training = sqlContext.createDataFrame(Seq(
    
23.       (1.218, 1.0, Vectors.dense(1.560, -0.605)),
    
24.       (2.949, 0.0, Vectors.dense(0.346, 2.158)),
    
25.       (3.627, 0.0, Vectors.dense(1.380, 0.231)),
    
26.       (0.273, 1.0, Vectors.dense(0.520, 1.151)),
    
27.       (4.199, 0.0, Vectors.dense(0.795, -0.226))
    
28.     )).toDF("label", "censor", "features")
    
29.     val quantileProbabilities = Array(0.3, 0.6)
    
30.     val aft = new AFTSurvivalRegression()
    
31.       .setQuantileProbabilities(quantileProbabilities)
    
32.       .setQuantilesCol("quantiles")
    
33. 
    
34.     val model = aft.fit(training)
    
35. 
    
36.     // Print the coefficients, intercept and scale parameter for AFT survival regression
    
37.     println(s"Coefficients: ${model.coefficients} Intercept: " +
    
38.       s"${model.intercept} Scale: ${model.scale}")
    
39.     model.transform(training).show(false)
    
40.     // $example off$
    
41. 
    
42.     sc.stop()
    
43.   }
    
44. }
    
45. // scalastyle:off println

复制代码

扫码加我 拉你入群

请注明：姓名-公司-职位

以便审核进群资格，未注明则拒绝

分享0 收藏0 回帖

关键词：regression Case study regressio Analysis Survival example import