By the way, I found in spark 2.1 I can use setFamily() to decide binomial or multinomial, but how can I do the same thing in spark 2.0.2? If not support , which one is used in spark 2.0.2? binomial or multinomial?
> 在 2017年3月19日,18:12,jinhong lu <lujinho...@gmail.com> 写道: > > > I train my LogisticRegressionModel like this, I want my model to retain only > some of the features(e.g. 500 of them), not all the 5555 features. What shou > I do? > I use .setElasticNetParam(1.0), but still all the features is in > lrModel.coefficients. > > import org.apache.spark.ml.classification.LogisticRegression > val > data=spark.read.format("libsvm").option("numFeatures","5555").load("/tmp/data/training_data3") > > val Array(trainingData, testData) = data.randomSplit(Array(0.5, 0.5), > seed = 1234L) > > val lr = new LogisticRegression() > val lrModel = lr.fit(trainingData) > println(s"Coefficients: ${lrModel.coefficients} Intercept: > ${lrModel.intercept}") > > val predictions = lrModel.transform(testData) > predictions.show() > > > Thanks, > lujinhong > Thanks, lujinhong