diff --git a/python/docs/pyspark.ml.rst b/python/docs/pyspark.ml.rst index a42217a4aee733e6e3148e05056ef290d42be414..8379b8fc8a1e1d81f8123aa978d54650e97a4b80 100644 --- a/python/docs/pyspark.ml.rst +++ b/python/docs/pyspark.ml.rst @@ -1,7 +1,7 @@ pyspark.ml package ===================== -Module Context +ML Pipeline APIs -------------- .. automodule:: pyspark.ml @@ -9,6 +9,14 @@ Module Context :undoc-members: :inherited-members: +pyspark.ml.param module +------------------------- + +.. automodule:: pyspark.ml.param + :members: + :undoc-members: + :inherited-members: + pyspark.ml.feature module ------------------------- diff --git a/python/pyspark/ml/__init__.py b/python/pyspark/ml/__init__.py index 47fed80f42e136a4271b326b90972ced659cef04..da793d9db7f918a10f617a46cf2580c76e3dcce6 100644 --- a/python/pyspark/ml/__init__.py +++ b/python/pyspark/ml/__init__.py @@ -15,7 +15,6 @@ # limitations under the License. # -from pyspark.ml.param import * -from pyspark.ml.pipeline import * +from pyspark.ml.pipeline import Transformer, Estimator, Model, Pipeline, PipelineModel, Evaluator -__all__ = ["Param", "Params", "Transformer", "Estimator", "Pipeline"] +__all__ = ["Transformer", "Estimator", "Model", "Pipeline", "PipelineModel", "Evaluator"] diff --git a/python/pyspark/ml/pipeline.py b/python/pyspark/ml/pipeline.py index fdbae06405f6a0345c3031b3d5541974ac0ada7f..a328bcf84a2e7c3282a6b55ec239e48af71705a3 100644 --- a/python/pyspark/ml/pipeline.py +++ b/python/pyspark/ml/pipeline.py @@ -22,9 +22,6 @@ from pyspark.ml.util import keyword_only from pyspark.mllib.common import inherit_doc -__all__ = ['Estimator', 'Transformer', 'Pipeline', 'PipelineModel', 'Evaluator', 'Model'] - - @inherit_doc class Estimator(Params): """