diff options
| author | Yanbo Liang <ybliang8@gmail.com> | 2015-08-17 23:57:02 -0700 |
|---|---|---|
| committer | Xiangrui Meng <meng@databricks.com> | 2015-08-17 23:57:02 -0700 |
| commit | dd0614fd618ad28cb77aecfbd49bb319b98fdba0 (patch) | |
| tree | a6c121c53cc133d293fbf289e959fd7718dceeda | |
| parent | 5af3838d2e59ed83766f85634e26918baa53819f (diff) | |
| download | spark-dd0614fd618ad28cb77aecfbd49bb319b98fdba0.tar.gz spark-dd0614fd618ad28cb77aecfbd49bb319b98fdba0.tar.bz2 spark-dd0614fd618ad28cb77aecfbd49bb319b98fdba0.zip | |
[SPARK-10076] [ML] make MultilayerPerceptronClassifier layers and weights public
Fix the issue that ```layers``` and ```weights``` should be public variables of ```MultilayerPerceptronClassificationModel```. Users can not get ```layers``` and ```weights``` from a ```MultilayerPerceptronClassificationModel``` currently.
Author: Yanbo Liang <ybliang8@gmail.com>
Closes #8263 from yanboliang/mlp-public.
| -rw-r--r-- | mllib/src/main/scala/org/apache/spark/ml/classification/MultilayerPerceptronClassifier.scala | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/mllib/src/main/scala/org/apache/spark/ml/classification/MultilayerPerceptronClassifier.scala b/mllib/src/main/scala/org/apache/spark/ml/classification/MultilayerPerceptronClassifier.scala index c154561886..ccca4ecc00 100644 --- a/mllib/src/main/scala/org/apache/spark/ml/classification/MultilayerPerceptronClassifier.scala +++ b/mllib/src/main/scala/org/apache/spark/ml/classification/MultilayerPerceptronClassifier.scala @@ -172,8 +172,8 @@ class MultilayerPerceptronClassifier(override val uid: String) @Experimental class MultilayerPerceptronClassificationModel private[ml] ( override val uid: String, - layers: Array[Int], - weights: Vector) + val layers: Array[Int], + val weights: Vector) extends PredictionModel[Vector, MultilayerPerceptronClassificationModel] with Serializable { |