Interface TreeEnsembleRegressorParams

All Superinterfaces:
DecisionTreeParams, HasCheckpointInterval, HasFeaturesCol, HasLabelCol, HasPredictionCol, HasSeed, HasWeightCol, Identifiable, Params, PredictorParams, Serializable, TreeEnsembleParams
All Known Subinterfaces:
GBTRegressorParams, RandomForestRegressorParams
All Known Implementing Classes:
GBTRegressionModel, GBTRegressor, RandomForestRegressionModel, RandomForestRegressor

public interface TreeEnsembleRegressorParams extends TreeEnsembleParams
Parameters for Decision Tree-based ensemble regression algorithms.
  • 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