Github user WeichenXu123 commented on a diff in the pull request:
https://github.com/apache/spark/pull/19172#discussion_r137922474
--- Diff: python/pyspark/ml/classification.py ---
@@ -1425,11 +1425,13 @@ class MultilayerPerceptronClassifier(JavaEstimator,
HasFeaturesCol, HasLabelCol,
@keyword_only
def __init__(self, featuresCol="features", labelCol="label",
predictionCol="prediction",
maxIter=100, tol=1e-6, seed=None, layers=None,
blockSize=128, stepSize=0.03,
- solver="l-bfgs", initialWeights=None):
+ solver="l-bfgs", initialWeights=None,
probabilityCol="probability",
+ rawPredicitionCol="rawPrediction"):
"""
__init__(self, featuresCol="features", labelCol="label",
predictionCol="prediction", \
maxIter=100, tol=1e-6, seed=None, layers=None,
blockSize=128, stepSize=0.03, \
- solver="l-bfgs", initialWeights=None)
+ solver="l-bfgs", initialWeights=None,
probabilityCol="probability", \
+ rawPredicitionCol="rawPrediction")
--- End diff --
Also add `probabilityCol="probability", rawPredicitionCol="rawPrediction"`
into the following line:
```self._setDefault(maxIter=100, tol=1E-6, blockSize=128, stepSize=0.03,
solver="l-bfgs")```
---
---------------------------------------------------------------------
To unsubscribe, e-mail: reviews-unsubscr...@spark.apache.org
For additional commands, e-mail: reviews-h...@spark.apache.org
