Interface ClassifierParams

All Superinterfaces:
HasFeaturesCol, HasLabelCol, HasPredictionCol, HasRawPredictionCol, Identifiable, Params, PredictorParams, Serializable, scala.Serializable
All Known Subinterfaces:
DecisionTreeClassifierParams, FMClassifierParams, GBTClassifierParams, LinearSVCParams, LogisticRegressionParams, MultilayerPerceptronParams, OneVsRestParams, ProbabilisticClassifierParams, RandomForestClassifierParams, TreeEnsembleClassifierParams
All Known Implementing Classes:
ClassificationModel, Classifier, DecisionTreeClassificationModel, DecisionTreeClassifier, FMClassificationModel, FMClassifier, GBTClassificationModel, GBTClassifier, LinearSVC, LinearSVCModel, LogisticRegression, LogisticRegressionModel, MultilayerPerceptronClassificationModel, MultilayerPerceptronClassifier, NaiveBayes, NaiveBayesModel, OneVsRest, OneVsRestModel, ProbabilisticClassificationModel, ProbabilisticClassifier, RandomForestClassificationModel, RandomForestClassifier

public interface ClassifierParams extends PredictorParams, HasRawPredictionCol
(private[spark]) Params for classification.
  • Method Details

    • validateAndTransformSchema

      StructType validateAndTransformSchema(StructType schema, boolean fitting, DataType featuresDataType)
      Description copied from interface: PredictorParams
      Validates and transforms the input schema with the provided param map.

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