.. currentmodule:: pyspark.ml

.. autosummary::
    :template: autosummary/class_with_docs.rst
    :toctree: api/

    Transformer
    UnaryTransformer
    Estimator
    Model
    Predictor
    PredictionModel
    Pipeline
    PipelineModel

.. currentmodule:: pyspark.ml.param

.. autosummary::
    :template: autosummary/class_with_docs.rst
    :toctree: api/

    Param
    Params
    TypeConverters

.. currentmodule:: pyspark.ml.feature

.. autosummary::
    :template: autosummary/class_with_docs.rst
    :toctree: api/

    Binarizer
    BucketedRandomProjectionLSH
    BucketedRandomProjectionLSHModel
    Bucketizer
    ChiSqSelector
    ChiSqSelectorModel
    CountVectorizer
    CountVectorizerModel
    DCT
    ElementwiseProduct
    FeatureHasher
    HashingTF
    IDF
    IDFModel
    Imputer
    ImputerModel
    IndexToString
    Interaction
    MaxAbsScaler
    MaxAbsScalerModel
    MinHashLSH
    MinHashLSHModel
    MinMaxScaler
    MinMaxScalerModel
    NGram
    Normalizer
    OneHotEncoder
    OneHotEncoderModel
    PCA
    PCAModel
    PolynomialExpansion
    QuantileDiscretizer
    RobustScaler
    RobustScalerModel
    RegexTokenizer
    RFormula
    RFormulaModel
    SQLTransformer
    StandardScaler
    StandardScalerModel
    StopWordsRemover
    StringIndexer
    StringIndexerModel
    Tokenizer
    UnivariateFeatureSelector
    UnivariateFeatureSelectorModel
    VarianceThresholdSelector
    VarianceThresholdSelectorModel
    VectorAssembler
    VectorIndexer
    VectorIndexerModel
    VectorSizeHint
    VectorSlicer
    Word2Vec
    Word2VecModel

.. currentmodule:: pyspark.ml.classification

.. autosummary::
    :template: autosummary/class_with_docs.rst
    :toctree: api/

    LinearSVC
    LinearSVCModel
    LinearSVCSummary
    LinearSVCTrainingSummary
    LogisticRegression
    LogisticRegressionModel
    LogisticRegressionSummary
    LogisticRegressionTrainingSummary
    BinaryLogisticRegressionSummary
    BinaryLogisticRegressionTrainingSummary
    DecisionTreeClassifier
    DecisionTreeClassificationModel
    GBTClassifier
    GBTClassificationModel
    RandomForestClassifier
    RandomForestClassificationModel
    RandomForestClassificationSummary
    RandomForestClassificationTrainingSummary
    BinaryRandomForestClassificationSummary
    BinaryRandomForestClassificationTrainingSummary
    NaiveBayes
    NaiveBayesModel
    MultilayerPerceptronClassifier
    MultilayerPerceptronClassificationModel
    MultilayerPerceptronClassificationSummary
    MultilayerPerceptronClassificationTrainingSummary
    OneVsRest
    OneVsRestModel
    FMClassifier
    FMClassificationModel
    FMClassificationSummary
    FMClassificationTrainingSummary

.. currentmodule:: pyspark.ml.clustering

.. autosummary::
    :template: autosummary/class_with_docs.rst
    :toctree: api/

    BisectingKMeans
    BisectingKMeansModel
    BisectingKMeansSummary
    KMeans
    KMeansModel
    KMeansSummary
    GaussianMixture
    GaussianMixtureModel
    GaussianMixtureSummary
    LDA
    LDAModel
    LocalLDAModel
    DistributedLDAModel
    PowerIterationClustering

.. currentmodule:: pyspark.ml.functions

.. autosummary::
    :toctree: api/

    array_to_vector
    vector_to_array
    predict_batch_udf

.. currentmodule:: pyspark.ml.linalg

.. autosummary::
    :template: autosummary/class_with_docs.rst
    :toctree: api/

    Vector
    DenseVector
    SparseVector
    Vectors
    Matrix
    DenseMatrix
    SparseMatrix
    Matrices

.. currentmodule:: pyspark.ml.recommendation

.. autosummary::
    :template: autosummary/class_with_docs.rst
    :toctree: api/

    ALS
    ALSModel

.. currentmodule:: pyspark.ml.regression

.. autosummary::
    :template: autosummary/class_with_docs.rst
    :toctree: api/

    AFTSurvivalRegression
    AFTSurvivalRegressionModel
    DecisionTreeRegressor
    DecisionTreeRegressionModel
    GBTRegressor
    GBTRegressionModel
    GeneralizedLinearRegression
    GeneralizedLinearRegressionModel
    GeneralizedLinearRegressionSummary
    GeneralizedLinearRegressionTrainingSummary
    IsotonicRegression
    IsotonicRegressionModel
    LinearRegression
    LinearRegressionModel
    LinearRegressionSummary
    LinearRegressionTrainingSummary
    RandomForestRegressor
    RandomForestRegressionModel
    FMRegressor
    FMRegressionModel

.. currentmodule:: pyspark.ml.stat

.. autosummary::
    :template: autosummary/class_with_docs.rst
    :toctree: api/

    ChiSquareTest
    Correlation
    KolmogorovSmirnovTest
    MultivariateGaussian
    Summarizer
    SummaryBuilder

.. currentmodule:: pyspark.ml.tuning

.. autosummary::
    :template: autosummary/class_with_docs.rst
    :toctree: api/

    ParamGridBuilder
    CrossValidator
    CrossValidatorModel
    TrainValidationSplit
    TrainValidationSplitModel

.. currentmodule:: pyspark.ml.evaluation

.. autosummary::
    :template: autosummary/class_with_docs.rst
    :toctree: api/

    Evaluator
    BinaryClassificationEvaluator
    RegressionEvaluator
    MulticlassClassificationEvaluator
    MultilabelClassificationEvaluator
    ClusteringEvaluator
    RankingEvaluator

.. currentmodule:: pyspark.ml.fpm

.. autosummary::
    :template: autosummary/class_with_docs.rst
    :toctree: api/

    FPGrowth
    FPGrowthModel
    PrefixSpan

.. currentmodule:: pyspark.ml.image

.. autosummary::
    :template: autosummary/class_with_docs.rst
    :toctree: api/

    ImageSchema
    _ImageSchema

.. currentmodule:: pyspark.ml.torch.distributor

.. autosummary::
    :template: autosummary/class_with_docs.rst
    :toctree: api/

    TorchDistributor

.. currentmodule:: pyspark.ml.deepspeed.deepspeed_distributor

.. autosummary::
    :template: autosummary/class_with_docs.rst
    :toctree: api/

    DeepspeedTorchDistributor

.. currentmodule:: pyspark.ml.util

.. autosummary::
    :template: autosummary/class_with_docs.rst
    :toctree: api/

    BaseReadWrite
    DefaultParamsReadable
    DefaultParamsReader
    DefaultParamsWritable
    DefaultParamsWriter
    GeneralMLWriter
    HasTrainingSummary
    Identifiable
    MLReadable
    MLReader
    MLWritable
    MLWriter