org.apache.spark.ml.tree

Interface RandomForestParams

All Superinterfaces:

DecisionTreeParams, HasCheckpointInterval, HasFeaturesCol, HasLabelCol, HasPredictionCol, HasSeed, Identifiable, Params, PredictorParams, java.io.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 Summary

Modifier and Type	Method and Description
int	getNumTrees()
IntParam	numTrees() Number of trees to train (at least 1).
RandomForestParams	setNumTrees(int value) Deprecated. This method is deprecated and will be removed in 3.0.0.

Methods inherited from interface org.apache.spark.ml.tree.TreeEnsembleParams

featureSubsetStrategy, getFeatureSubsetStrategy, getOldStrategy, getSubsamplingRate, setFeatureSubsetStrategy, setSubsamplingRate, subsamplingRate

Methods inherited from interface org.apache.spark.ml.tree.DecisionTreeParams

cacheNodeIds, getCacheNodeIds, getMaxBins, getMaxDepth, getMaxMemoryInMB, getMinInfoGain, getMinInstancesPerNode, getOldStrategy, maxBins, maxDepth, maxMemoryInMB, minInfoGain, minInstancesPerNode, setCacheNodeIds, setCheckpointInterval, setMaxBins, setMaxDepth, setMaxMemoryInMB, setMinInfoGain, setMinInstancesPerNode, setSeed

Methods inherited from interface org.apache.spark.ml.PredictorParams

validateAndTransformSchema

Methods inherited from interface org.apache.spark.ml.param.shared.HasLabelCol

getLabelCol, labelCol

Methods inherited from interface org.apache.spark.ml.param.shared.HasFeaturesCol

featuresCol, getFeaturesCol

Methods inherited from interface org.apache.spark.ml.param.shared.HasPredictionCol

getPredictionCol, predictionCol

Methods inherited from interface org.apache.spark.ml.param.Params

clear, copy, copyValues, defaultCopy, defaultParamMap, explainParam, explainParams, extractParamMap, extractParamMap, get, getDefault, getOrDefault, getParam, hasDefault, hasParam, isDefined, isSet, paramMap, params, set, set, set, setDefault, setDefault, shouldOwn

Methods inherited from interface org.apache.spark.ml.util.Identifiable

toString, uid

Methods inherited from interface org.apache.spark.ml.param.shared.HasCheckpointInterval

checkpointInterval, getCheckpointInterval

Methods inherited from interface org.apache.spark.ml.param.shared.HasSeed

getSeed, seed

Method Detail

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.

Returns:

(undocumented)

setNumTrees

RandomForestParams setNumTrees(int value)

Deprecated. This method is deprecated and will be removed in 3.0.0.

Parameters:

value - (undocumented)

Returns:

(undocumented)