[Case Study]AFT Survival Regression using Java

是 否 +2 论坛币

k人 参与回答

经管之家送您一份

应届毕业生专属福利!

求职就业群

赵安豆老师微信：zhaoandou666

经管之家联合CDA

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

立即领取

感谢您参与论坛问题回答

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

+2 论坛币

1. package org.apache.spark.examples.ml;
   
2. 
   
3. // $example on$
   
4. import java.util.Arrays;
   
5. import java.util.List;
   
6. 
   
7. import org.apache.spark.SparkConf;
   
8. import org.apache.spark.api.java.JavaSparkContext;
   
9. import org.apache.spark.ml.regression.AFTSurvivalRegression;
   
10. import org.apache.spark.ml.regression.AFTSurvivalRegressionModel;
    
11. import org.apache.spark.mllib.linalg.*;
    
12. import org.apache.spark.sql.DataFrame;
    
13. import org.apache.spark.sql.Row;
    
14. import org.apache.spark.sql.RowFactory;
    
15. import org.apache.spark.sql.SQLContext;
    
16. import org.apache.spark.sql.types.*;
    
17. // $example off$
    
18. 
    
19. public class JavaAFTSurvivalRegressionExample {
    
20.   public static void main(String[] args) {
    
21.     SparkConf conf = new SparkConf().setAppName("JavaAFTSurvivalRegressionExample");
    
22.     JavaSparkContext jsc = new JavaSparkContext(conf);
    
23.     SQLContext jsql = new SQLContext(jsc);
    
24. 
    
25.     // $example on$
    
26.     List<Row> data = Arrays.asList(
    
27.       RowFactory.create(1.218, 1.0, Vectors.dense(1.560, -0.605)),
    
28.       RowFactory.create(2.949, 0.0, Vectors.dense(0.346, 2.158)),
    
29.       RowFactory.create(3.627, 0.0, Vectors.dense(1.380, 0.231)),
    
30.       RowFactory.create(0.273, 1.0, Vectors.dense(0.520, 1.151)),
    
31.       RowFactory.create(4.199, 0.0, Vectors.dense(0.795, -0.226))
    
32.     );
    
33.     StructType schema = new StructType(new StructField[]{
    
34.       new StructField("label", DataTypes.DoubleType, false, Metadata.empty()),
    
35.       new StructField("censor", DataTypes.DoubleType, false, Metadata.empty()),
    
36.       new StructField("features", new VectorUDT(), false, Metadata.empty())
    
37.     });
    
38.     DataFrame training = jsql.createDataFrame(data, schema);
    
39.     double[] quantileProbabilities = new double[]{0.3, 0.6};
    
40.     AFTSurvivalRegression aft = new AFTSurvivalRegression()
    
41.       .setQuantileProbabilities(quantileProbabilities)
    
42.       .setQuantilesCol("quantiles");
    
43. 
    
44.     AFTSurvivalRegressionModel model = aft.fit(training);
    
45. 
    
46.     // Print the coefficients, intercept and scale parameter for AFT survival regression
    
47.     System.out.println("Coefficients: " + model.coefficients() + " Intercept: "
    
48.       + model.intercept() + " Scale: " + model.scale());
    
49.     model.transform(training).show(false);
    
50.     // $example off$
    
51. 
    
52.     jsc.stop();
    
53.   }
    
54. }

复制代码

扫码加我 拉你入群

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

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

分享0 收藏0 回帖

关键词：regression Case study regressio Survival regress example package import Java