Interface PredictorParams

All Superinterfaces:
HasFeaturesCol, HasLabelCol, HasPredictionCol, Identifiable, Params, Serializable
All Known Subinterfaces:
AFTSurvivalRegressionParams, ClassifierParams, DecisionTreeClassifierParams, DecisionTreeParams, DecisionTreeRegressorParams, FactorizationMachines, FactorizationMachinesParams, FMClassifierParams, FMRegressorParams, GBTClassifierParams, GBTParams, GBTRegressorParams, GeneralizedLinearRegressionBase, LinearRegressionParams, LinearSVCParams, LogisticRegressionParams, MultilayerPerceptronParams, NaiveBayesParams, OneVsRestParams, ProbabilisticClassifierParams, RandomForestClassifierParams, RandomForestParams, RandomForestRegressorParams, TreeEnsembleClassifierParams, TreeEnsembleParams, TreeEnsembleRegressorParams
All Known Implementing Classes:
AFTSurvivalRegression, AFTSurvivalRegressionModel, ClassificationModel, Classifier, DecisionTreeClassificationModel, DecisionTreeClassifier, DecisionTreeRegressionModel, DecisionTreeRegressor, FMClassificationModel, FMClassifier, FMRegressionModel, FMRegressor, GBTClassificationModel, GBTClassifier, GBTRegressionModel, GBTRegressor, GeneralizedLinearRegression, GeneralizedLinearRegressionModel, LinearRegression, LinearRegressionModel, LinearSVC, LinearSVCModel, LogisticRegression, LogisticRegressionModel, MultilayerPerceptronClassificationModel, MultilayerPerceptronClassifier, NaiveBayes, NaiveBayesModel, OneVsRest, OneVsRestModel, PredictionModel, Predictor, ProbabilisticClassificationModel, ProbabilisticClassifier, RandomForestClassificationModel, RandomForestClassifier, RandomForestRegressionModel, RandomForestRegressor, RegressionModel, Regressor

public interface PredictorParams extends Params, HasLabelCol, HasFeaturesCol, HasPredictionCol
(private[ml]) Trait for parameters for prediction (regression and classification).
  • Method Details

    • validateAndTransformSchema

      StructType validateAndTransformSchema(StructType schema, boolean fitting, DataType featuresDataType)
      Validates and transforms the input schema with the provided param map.

      Parameters:
      schema - input schema
      fitting - whether this is in fitting
      featuresDataType - SQL DataType for FeaturesType. E.g., VectorUDT for vector features.
      Returns:
      output schema