diff options
| author | Yanbo Liang <ybliang8@gmail.com> | 2016-09-06 03:30:37 -0700 |
|---|---|---|
| committer | Yanbo Liang <ybliang8@gmail.com> | 2016-09-06 03:30:37 -0700 |
| commit | 39d538dddf7d44bf4603c966d0f7b2c92f1e951a (patch) | |
| tree | 3a172259e76f88f66098644efa4d57336bea42e1 | |
| parent | 6f13aa7dfee12b1b301bd10a1050549008ecc67e (diff) | |
| download | spark-39d538dddf7d44bf4603c966d0f7b2c92f1e951a.tar.gz spark-39d538dddf7d44bf4603c966d0f7b2c92f1e951a.tar.bz2 spark-39d538dddf7d44bf4603c966d0f7b2c92f1e951a.zip | |
[MINOR][ML] Correct weights doc of MultilayerPerceptronClassificationModel.
## What changes were proposed in this pull request?
```weights``` of ```MultilayerPerceptronClassificationModel``` should be the output weights of layers rather than initial weights, this PR correct it.
## How was this patch tested?
Doc change.
Author: Yanbo Liang <ybliang8@gmail.com>
Closes #14967 from yanboliang/mlp-weights.
| -rw-r--r-- | mllib/src/main/scala/org/apache/spark/ml/classification/MultilayerPerceptronClassifier.scala | 2 | ||||
| -rw-r--r-- | python/pyspark/ml/classification.py | 2 |
2 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 7264a99b47..88fe7cb4a6 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 @@ -288,7 +288,7 @@ object MultilayerPerceptronClassifier * * @param uid uid * @param layers array of layer sizes including input and output layers - * @param weights vector of initial weights for the model that consists of the weights of layers + * @param weights the weights of layers * @return prediction model */ @Since("1.5.0") diff --git a/python/pyspark/ml/classification.py b/python/pyspark/ml/classification.py index d1522d78fa..b4c01fd5c4 100644 --- a/python/pyspark/ml/classification.py +++ b/python/pyspark/ml/classification.py @@ -1276,7 +1276,7 @@ class MultilayerPerceptronClassificationModel(JavaModel, JavaPredictionModel, Ja @since("2.0.0") def weights(self): """ - vector of initial weights for the model that consists of the weights of layers. + the weights of layers. """ return self._call_java("weights") |