Skip to content

Commit dd0614f

Browse files
yanboliangmengxr
authored andcommitted
[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 <[email protected]> Closes #8263 from yanboliang/mlp-public.
1 parent 5af3838 commit dd0614f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

mllib/src/main/scala/org/apache/spark/ml/classification/MultilayerPerceptronClassifier.scala

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -172,8 +172,8 @@ class MultilayerPerceptronClassifier(override val uid: String)
172172
@Experimental
173173
class MultilayerPerceptronClassificationModel private[ml] (
174174
override val uid: String,
175-
layers: Array[Int],
176-
weights: Vector)
175+
val layers: Array[Int],
176+
val weights: Vector)
177177
extends PredictionModel[Vector, MultilayerPerceptronClassificationModel]
178178
with Serializable {
179179

0 commit comments

Comments
 (0)