Address huaxingao comments

apache · Feb 19, 2020 · 0b0f723 · 0b0f723
1 parent 63e2fcc
commit 0b0f723
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/python/pyspark/ml/__init__.py b/python/pyspark/ml/__init__.py
@@ -27,8 +27,7 @@
 
 __all__ = [
     "Transformer", "UnaryTransformer", "Estimator", "Model",
-    "Predictor", "PredictionModel",
-    "Pipeline", "PipelineModel",
+    "Predictor", "PredictionModel", "Pipeline", "PipelineModel",
     "classification", "clustering", "evaluation", "feature", "fpm", "image",
     "recommendation", "regression", "stat", "tuning", "util", "linalg", "param",
 ]
diff --git a/python/pyspark/ml/classification.py b/python/pyspark/ml/classification.py
@@ -112,8 +112,8 @@ def predictRaw(self, value):
 
 class _ProbabilisticClassifierParams(HasProbabilityCol, HasThresholds, _ClassifierParams):
     """
-    Params for :py:class:`_JavaProbabilisticClassifier` and
-    :py:class:`_JavaProbabilisticClassificationModel`.
+    Params for :py:class:`ProbabilisticClassifier` and
+    :py:class:`ProbabilisticClassificationModel`.
 
     .. versionadded:: 3.0.0
     """