-
- Downloads
[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 <holden@us.ibm.com> Closes #12889 from holdenk/SPARK-15113-add-missing-numFeatures-numClasses.
Showing
- mllib/src/main/scala/org/apache/spark/ml/regression/GeneralizedLinearRegression.scala 2 additions, 0 deletions...che/spark/ml/regression/GeneralizedLinearRegression.scala
- python/pyspark/ml/classification.py 31 additions, 6 deletionspython/pyspark/ml/classification.py
- python/pyspark/ml/regression.py 17 additions, 5 deletionspython/pyspark/ml/regression.py
- python/pyspark/ml/util.py 16 additions, 0 deletionspython/pyspark/ml/util.py
Loading
Please register or sign in to comment