Interface RandomForestParams

All Superinterfaces:
DecisionTreeParams, HasCheckpointInterval, HasFeaturesCol, HasLabelCol, HasPredictionCol, HasSeed, HasWeightCol, Identifiable, Params, PredictorParams, Serializable, scala.Serializable, TreeEnsembleParams
All Known Subinterfaces:
RandomForestClassifierParams, RandomForestRegressorParams
All Known Implementing Classes:
RandomForestClassificationModel, RandomForestClassifier, RandomForestRegressionModel, RandomForestRegressor

public interface RandomForestParams extends TreeEnsembleParams
Parameters for Random Forest algorithms.
  • Method Details

    • bootstrap

      BooleanParam bootstrap()
      Whether bootstrap samples are used when building trees.
    • getBootstrap

      boolean getBootstrap()
    • getNumTrees

      int getNumTrees()
    • numTrees

      IntParam numTrees()
      Number of trees to train (at least 1). If 1, then no bootstrapping is used. If greater than 1, then bootstrapping is done. TODO: Change to always do bootstrapping (simpler). SPARK-7130 (default = 20)

      Note: The reason that we cannot add this to both GBT and RF (i.e. in TreeEnsembleParams) is the param maxIter controls how many trees a GBT has. The semantics in the algorithms are a bit different.