pyspark/ml/__init__.py

0001 #
0002 # Licensed to the Apache Software Foundation (ASF) under one or more
0003 # contributor license agreements.  See the NOTICE file distributed with
0004 # this work for additional information regarding copyright ownership.
0005 # The ASF licenses this file to You under the Apache License, Version 2.0
0006 # (the "License"); you may not use this file except in compliance with
0007 # the License.  You may obtain a copy of the License at
0008 #
0009 #    http://www.apache.org/licenses/LICENSE-2.0
0010 #
0011 # Unless required by applicable law or agreed to in writing, software
0012 # distributed under the License is distributed on an "AS IS" BASIS,
0013 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
0014 # See the License for the specific language governing permissions and
0015 # limitations under the License.
0016 #
0017
0018 """
0019 DataFrame-based machine learning APIs to let users quickly assemble and configure practical
0020 machine learning pipelines.
0021 """
0022 from pyspark.ml.base import Estimator, Model, Transformer, UnaryTransformer
0023 from pyspark.ml.pipeline import Pipeline, PipelineModel
0024 from pyspark.ml import classification, clustering, evaluation, feature, fpm, \
0025     image, pipeline, recommendation, regression, stat, tuning, util, linalg, param
0026
0027 __all__ = [
0028     "Transformer", "UnaryTransformer", "Estimator", "Model", "Pipeline", "PipelineModel",
0029     "classification", "clustering", "evaluation", "feature", "fpm", "image",
0030     "recommendation", "regression", "stat", "tuning", "util", "linalg", "param",
0031 ]