Github user mengxr commented on a diff in the pull request:
https://github.com/apache/spark/pull/21483#discussion_r193785341
--- Diff: python/pyspark/ml/__init__.py ---
@@ -21,5 +21,11 @@
"""
from pyspark.ml.base import Estimator, Model, Transformer, UnaryTransformer
from pyspark.ml.pipeline import Pipeline, PipelineModel
+from pyspark.ml import classification, clustering, evaluation, feature,
fpm, \
+ image, pipeline, recommendation, regression, stat, tuning, util
-__all__ = ["Transformer", "UnaryTransformer", "Estimator", "Model",
"Pipeline", "PipelineModel"]
+__all__ = [
+ "Transformer", "UnaryTransformer", "Estimator", "Model", "Pipeline",
"PipelineModel",
+ "base", "classification", "clustering", "evaluation", "feature", "fpm",
+ "image", "pipeline", "recommendation", "regression", "stat", "tuning",
"util",
--- End diff --
we should remove `pipeline` from default imports and add `linalg` and
`param`. Basically, import public modules listed here:
https://spark.apache.org/docs/latest/api/python/index.html.
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]