From b264cbb16fb97116e630fb593adf5898a5a0e8fa Mon Sep 17 00:00:00 2001 From: Holden Karau Date: Mon, 22 Aug 2016 12:21:22 +0200 Subject: [SPARK-15113][PYSPARK][ML] Add missing num features num classes ## What changes were proposed in this pull request? Add missing `numFeatures` and `numClasses` to the wrapped Java models in PySpark ML pipelines. Also tag `DecisionTreeClassificationModel` as Expiremental to match Scala doc. ## How was this patch tested? Extended doctests Author: Holden Karau Closes #12889 from holdenk/SPARK-15113-add-missing-numFeatures-numClasses. --- .../org/apache/spark/ml/regression/GeneralizedLinearRegression.scala | 2 ++ 1 file changed, 2 insertions(+) (limited to 'mllib/src') diff --git a/mllib/src/main/scala/org/apache/spark/ml/regression/GeneralizedLinearRegression.scala b/mllib/src/main/scala/org/apache/spark/ml/regression/GeneralizedLinearRegression.scala index 2bdc09e1db..1d4dfd1147 100644 --- a/mllib/src/main/scala/org/apache/spark/ml/regression/GeneralizedLinearRegression.scala +++ b/mllib/src/main/scala/org/apache/spark/ml/regression/GeneralizedLinearRegression.scala @@ -788,6 +788,8 @@ class GeneralizedLinearRegressionModel private[ml] ( @Since("2.0.0") override def write: MLWriter = new GeneralizedLinearRegressionModel.GeneralizedLinearRegressionModelWriter(this) + + override val numFeatures: Int = coefficients.size } @Since("2.0.0") -- cgit v1.2.3