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

Reply via email to