estimators

Type Members

class AFTSurvivalRegression extends SparkEstimatorWrapper[AFTSurvivalRegressionModel, org.apache.spark.ml.regression.AFTSurvivalRegression, AFTSurvivalRegressionModel] with AFTSurvivalRegressionParams with HasLabelColumnParam with HasMaxIterationsParam with HasTolerance with HasFitIntercept
class ALS extends SparkEstimatorWrapper[ALSModel, org.apache.spark.ml.recommendation.ALS, ALSModel] with HasItemColumnParam with HasPredictionColumnCreatorParam with HasUserColumnParam with HasMaxIterationsParam with HasSeedParam with HasRegularizationParam with HasCheckpointIntervalParam
class ChiSqSelectorEstimator extends SparkEstimatorWrapper[ChiSqSelectorModel, ChiSqSelector, ChiSqSelectorModel] with HasFeaturesColumnParam with HasOutputColumn with HasLabelColumnParam
class CountVectorizerEstimator extends SparkSingleColumnEstimatorWrapper[CountVectorizerModel, CountVectorizer, CountVectorizerModel] with HasMinTermsFrequencyParam
class DecisionTreeClassifier extends StringIndexingEstimatorWrapper[DecisionTreeClassificationModel, org.apache.spark.ml.classification.DecisionTreeClassifier, VanillaDecisionTreeClassificationModel, DecisionTreeClassificationModel]
class DecisionTreeRegression extends SparkEstimatorWrapper[DecisionTreeRegressionModel, DecisionTreeRegressor, DecisionTreeRegressionModel] with DecisionTreeParams with HasRegressionImpurityParam with HasLabelColumnParam
class GBTClassifier extends StringIndexingEstimatorWrapper[GBTClassificationModel, org.apache.spark.ml.classification.GBTClassifier, VanillaGBTClassificationModel, GBTClassificationModel]
class GBTRegression extends SparkEstimatorWrapper[GBTRegressionModel, GBTRegressor, GBTRegressionModel] with GBTParams with HasRegressionImpurityParam
class IDFEstimator extends SparkSingleColumnEstimatorWrapper[IDFModel, IDF, IDFModel]
class IsotonicRegression extends SparkEstimatorWrapper[IsotonicRegressionModel, org.apache.spark.ml.regression.IsotonicRegression, IsotonicRegressionModel] with PredictorParams with HasFeatureIndexParam with HasLabelColumnParam with HasOptionalWeightColumnParam
class KMeans extends SparkEstimatorWrapper[KMeansModel, org.apache.spark.ml.clustering.KMeans, KMeansModel] with PredictorParams with HasNumberOfClustersParam with HasMaxIterationsParam with HasSeedParam with HasTolerance
class LDA extends SparkEstimatorWrapper[LDAModel, org.apache.spark.ml.clustering.LDA, LDAModel] with HasCheckpointIntervalParam with HasFeaturesColumnParam with HasNumberOfClustersParam with HasMaxIterationsParam with HasSeedParam
class LinearRegression extends SparkEstimatorWrapper[LinearRegressionModel, org.apache.spark.ml.regression.LinearRegression, LinearRegressionModel] with LinearRegressionParams
class LogisticRegression extends SparkEstimatorWrapper[LogisticRegressionModel, org.apache.spark.ml.classification.LogisticRegression, LogisticRegressionModel] with ProbabilisticClassifierParams with HasLabelColumnParam with HasThreshold with HasRegularizationParam with HasElasticNetParam with HasMaxIterationsParam with HasTolerance with HasFitIntercept with HasStandardization with HasOptionalWeightColumnParam
class MinMaxScalerEstimator extends SparkSingleColumnEstimatorWrapper[MinMaxScalerModel, MinMaxScaler, MinMaxScalerModel] with MinMaxParams with HasInputColumn with HasOutputColumn
class MultilayerPerceptronClassifier extends SparkEstimatorWrapper[MultilayerPerceptronClassificationModel, org.apache.spark.ml.classification.MultilayerPerceptronClassifier, MultilayerPerceptronClassifierModel] with PredictorParams with HasLabelColumnParam with HasMaxIterationsParam with HasSeedParam with HasTolerance
class NaiveBayes extends SparkEstimatorWrapper[NaiveBayesModel, org.apache.spark.ml.classification.NaiveBayes, NaiveBayesModel] with ProbabilisticClassifierParams with HasLabelColumnParam
class PCAEstimator extends SparkSingleColumnEstimatorWrapper[PCAModel, PCA, PCAModel]
class QuantileDiscretizerEstimator extends SparkSingleColumnEstimatorWrapper[Bucketizer, QuantileDiscretizer, QuantileDiscretizerModel]
class RandomForestClassifier extends StringIndexingEstimatorWrapper[RandomForestClassificationModel, org.apache.spark.ml.classification.RandomForestClassifier, VanillaRandomForestClassificationModel, RandomForestClassificationModel]
class RandomForestRegression extends SparkEstimatorWrapper[RandomForestRegressionModel, RandomForestRegressor, RandomForestRegressionModel] with PredictorParams with HasLabelColumnParam with HasSeedParam with HasMaxDepthParam with HasMinInstancePerNodeParam with HasMaxBinsParam with HasSubsamplingRateParam with HasMinInfoGainParam with HasMaxMemoryInMBParam with HasCacheNodeIdsParam with HasCheckpointIntervalParam with HasNumTreesParam with HasFeatureSubsetStrategyParam with HasRegressionImpurityParam with Logging
class SingleStringIndexer extends SparkSingleColumnEstimatorWrapper[StringIndexerModel, StringIndexer, SingleColumnStringIndexerModel]
class StandardScalerEstimator extends SparkSingleColumnEstimatorWrapper[StandardScalerModel, StandardScaler, StandardScalerModel]
class StringIndexerEstimator extends SparkMultiColumnEstimatorWrapper[StringIndexerModel, StringIndexer, StringIndexerModel, SingleColumnStringIndexerModel, SingleStringIndexer, MultiColumnStringIndexerModel]
class VanillaDecisionTreeClassifier extends SparkEstimatorWrapper[DecisionTreeClassificationModel, org.apache.spark.ml.classification.DecisionTreeClassifier, VanillaDecisionTreeClassificationModel] with HasClassificationImpurityParam with DecisionTreeParams with ProbabilisticClassifierParams with HasLabelColumnParam
class VanillaGBTClassifier extends SparkEstimatorWrapper[GBTClassificationModel, org.apache.spark.ml.classification.GBTClassifier, VanillaGBTClassificationModel] with GBTParams with HasClassificationImpurityParam with Logging
class VanillaRandomForestClassifier extends SparkEstimatorWrapper[RandomForestClassificationModel, org.apache.spark.ml.classification.RandomForestClassifier, VanillaRandomForestClassificationModel] with HasMaxDepthParam with HasMaxBinsParam with HasMinInstancePerNodeParam with HasMinInfoGainParam with HasMaxMemoryInMBParam with HasCacheNodeIdsParam with HasCheckpointIntervalParam with HasSubsamplingRateParam with HasSeedParam with HasNumTreesParam with HasFeatureSubsetStrategyParam with PredictorParams with HasLabelColumnParam with ProbabilisticClassifierParams with HasClassificationImpurityParam
class VectorIndexerEstimator extends SparkSingleColumnEstimatorWrapper[VectorIndexerModel, VectorIndexer, VectorIndexerModel]
class Word2VecEstimator extends SparkSingleColumnEstimatorWrapper[Word2VecModel, Word2Vec, Word2VecModel] with Word2VecParams

Value Members

object GBTClassifier extends Serializable
object GBTRegression extends Serializable
object KMeans extends Serializable
object LDA extends Serializable
object NaiveBayes extends Serializable

package estimators

Type Members

class AFTSurvivalRegression extends SparkEstimatorWrapper[AFTSurvivalRegressionModel, org.apache.spark.ml.regression.AFTSurvivalRegression, AFTSurvivalRegressionModel] with AFTSurvivalRegressionParams with HasLabelColumnParam with HasMaxIterationsParam with HasTolerance with HasFitIntercept

class ALS extends SparkEstimatorWrapper[ALSModel, org.apache.spark.ml.recommendation.ALS, ALSModel] with HasItemColumnParam with HasPredictionColumnCreatorParam with HasUserColumnParam with HasMaxIterationsParam with HasSeedParam with HasRegularizationParam with HasCheckpointIntervalParam

class ChiSqSelectorEstimator extends SparkEstimatorWrapper[ChiSqSelectorModel, ChiSqSelector, ChiSqSelectorModel] with HasFeaturesColumnParam with HasOutputColumn with HasLabelColumnParam

class CountVectorizerEstimator extends SparkSingleColumnEstimatorWrapper[CountVectorizerModel, CountVectorizer, CountVectorizerModel] with HasMinTermsFrequencyParam

class DecisionTreeClassifier extends StringIndexingEstimatorWrapper[DecisionTreeClassificationModel, org.apache.spark.ml.classification.DecisionTreeClassifier, VanillaDecisionTreeClassificationModel, DecisionTreeClassificationModel]

class DecisionTreeRegression extends SparkEstimatorWrapper[DecisionTreeRegressionModel, DecisionTreeRegressor, DecisionTreeRegressionModel] with DecisionTreeParams with HasRegressionImpurityParam with HasLabelColumnParam

class GBTClassifier extends StringIndexingEstimatorWrapper[GBTClassificationModel, org.apache.spark.ml.classification.GBTClassifier, VanillaGBTClassificationModel, GBTClassificationModel]

class GBTRegression extends SparkEstimatorWrapper[GBTRegressionModel, GBTRegressor, GBTRegressionModel] with GBTParams with HasRegressionImpurityParam

class IDFEstimator extends SparkSingleColumnEstimatorWrapper[IDFModel, IDF, IDFModel]

class IsotonicRegression extends SparkEstimatorWrapper[IsotonicRegressionModel, org.apache.spark.ml.regression.IsotonicRegression, IsotonicRegressionModel] with PredictorParams with HasFeatureIndexParam with HasLabelColumnParam with HasOptionalWeightColumnParam

class KMeans extends SparkEstimatorWrapper[KMeansModel, org.apache.spark.ml.clustering.KMeans, KMeansModel] with PredictorParams with HasNumberOfClustersParam with HasMaxIterationsParam with HasSeedParam with HasTolerance

class LDA extends SparkEstimatorWrapper[LDAModel, org.apache.spark.ml.clustering.LDA, LDAModel] with HasCheckpointIntervalParam with HasFeaturesColumnParam with HasNumberOfClustersParam with HasMaxIterationsParam with HasSeedParam

class LinearRegression extends SparkEstimatorWrapper[LinearRegressionModel, org.apache.spark.ml.regression.LinearRegression, LinearRegressionModel] with LinearRegressionParams

class MinMaxScalerEstimator extends SparkSingleColumnEstimatorWrapper[MinMaxScalerModel, MinMaxScaler, MinMaxScalerModel] with MinMaxParams with HasInputColumn with HasOutputColumn

class NaiveBayes extends SparkEstimatorWrapper[NaiveBayesModel, org.apache.spark.ml.classification.NaiveBayes, NaiveBayesModel] with ProbabilisticClassifierParams with HasLabelColumnParam

class PCAEstimator extends SparkSingleColumnEstimatorWrapper[PCAModel, PCA, PCAModel]

class QuantileDiscretizerEstimator extends SparkSingleColumnEstimatorWrapper[Bucketizer, QuantileDiscretizer, QuantileDiscretizerModel]

class RandomForestClassifier extends StringIndexingEstimatorWrapper[RandomForestClassificationModel, org.apache.spark.ml.classification.RandomForestClassifier, VanillaRandomForestClassificationModel, RandomForestClassificationModel]

class SingleStringIndexer extends SparkSingleColumnEstimatorWrapper[StringIndexerModel, StringIndexer, SingleColumnStringIndexerModel]

class StandardScalerEstimator extends SparkSingleColumnEstimatorWrapper[StandardScalerModel, StandardScaler, StandardScalerModel]

class StringIndexerEstimator extends SparkMultiColumnEstimatorWrapper[StringIndexerModel, StringIndexer, StringIndexerModel, SingleColumnStringIndexerModel, SingleStringIndexer, MultiColumnStringIndexerModel]

class VanillaGBTClassifier extends SparkEstimatorWrapper[GBTClassificationModel, org.apache.spark.ml.classification.GBTClassifier, VanillaGBTClassificationModel] with GBTParams with HasClassificationImpurityParam with Logging

class VectorIndexerEstimator extends SparkSingleColumnEstimatorWrapper[VectorIndexerModel, VectorIndexer, VectorIndexerModel]

class Word2VecEstimator extends SparkSingleColumnEstimatorWrapper[Word2VecModel, Word2Vec, Word2VecModel] with Word2VecParams

Value Members

object GBTClassifier extends Serializable

object GBTRegression extends Serializable

object KMeans extends Serializable

object LDA extends Serializable

object NaiveBayes extends Serializable

Ungrouped