pyspark.ml
Transformer UnaryTransformer Estimator Model Predictor PredictionModel Pipeline PipelineModel
pyspark.ml.param
Param Params TypeConverters
pyspark.ml.feature
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
pyspark.ml.classification
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
pyspark.ml.clustering
BisectingKMeans BisectingKMeansModel BisectingKMeansSummary KMeans KMeansModel KMeansSummary GaussianMixture GaussianMixtureModel GaussianMixtureSummary LDA LDAModel LocalLDAModel DistributedLDAModel PowerIterationClustering
pyspark.ml.functions
array_to_vector vector_to_array predict_batch_udf
pyspark.ml.linalg
Vector DenseVector SparseVector Vectors Matrix DenseMatrix SparseMatrix Matrices
pyspark.ml.recommendation
ALS ALSModel
pyspark.ml.regression
AFTSurvivalRegression AFTSurvivalRegressionModel DecisionTreeRegressor DecisionTreeRegressionModel GBTRegressor GBTRegressionModel GeneralizedLinearRegression GeneralizedLinearRegressionModel GeneralizedLinearRegressionSummary GeneralizedLinearRegressionTrainingSummary IsotonicRegression IsotonicRegressionModel LinearRegression LinearRegressionModel LinearRegressionSummary LinearRegressionTrainingSummary RandomForestRegressor RandomForestRegressionModel FMRegressor FMRegressionModel
pyspark.ml.stat
ChiSquareTest Correlation KolmogorovSmirnovTest MultivariateGaussian Summarizer SummaryBuilder
pyspark.ml.tuning
ParamGridBuilder CrossValidator CrossValidatorModel TrainValidationSplit TrainValidationSplitModel
pyspark.ml.evaluation
Evaluator BinaryClassificationEvaluator RegressionEvaluator MulticlassClassificationEvaluator MultilabelClassificationEvaluator ClusteringEvaluator RankingEvaluator
pyspark.ml.fpm
FPGrowth FPGrowthModel PrefixSpan
pyspark.ml.image
ImageSchema _ImageSchema
pyspark.ml.torch.distributor
TorchDistributor
pyspark.ml.util
BaseReadWrite DefaultParamsReadable DefaultParamsReader DefaultParamsWritable DefaultParamsWriter GeneralMLWriter HasTrainingSummary Identifiable MLReadable MLReader MLWritable MLWriter