-
- Downloads
[SPARK-12664][ML] Expose probability in mlp model
## What changes were proposed in this pull request? Modify MLP model to inherit `ProbabilisticClassificationModel` and so that it can expose the probability column when transforming data. ## How was this patch tested? Test added. Author: WeichenXu <WeichenXu123@outlook.com> Closes #17373 from WeichenXu123/expose_probability_in_mlp_model.
Showing
- mllib/src/main/scala/org/apache/spark/ml/ann/Layer.scala 45 additions, 8 deletionsmllib/src/main/scala/org/apache/spark/ml/ann/Layer.scala
- mllib/src/main/scala/org/apache/spark/ml/classification/MultilayerPerceptronClassifier.scala 13 additions, 4 deletions...rk/ml/classification/MultilayerPerceptronClassifier.scala
- mllib/src/test/scala/org/apache/spark/ml/ann/GradientSuite.scala 1 addition, 1 deletion...rc/test/scala/org/apache/spark/ml/ann/GradientSuite.scala
- mllib/src/test/scala/org/apache/spark/ml/classification/MultilayerPerceptronClassifierSuite.scala 42 additions, 0 deletions.../classification/MultilayerPerceptronClassifierSuite.scala
- python/pyspark/ml/classification.py 2 additions, 2 deletionspython/pyspark/ml/classification.py
Loading
Please register or sign in to comment