case class RandomForest(numTrees: Int = -1, useJackknife: Boolean = true, biasLearner: Option[Learner] = None, leafLearner: Option[Learner] = None, subsetStrategy: Any = "auto", minLeafInstances: Int = 1, maxDepth: Int = Integer.MAX_VALUE, uncertaintyCalibration: Boolean = true, randomizePivotLocation: Boolean = false, randomlyRotateFeatures: Boolean = false, rng: Random = Random) extends Learner with Product with Serializable
Standard random forest as a wrapper around bagged decision trees Created by maxhutch on 1/9/17.
- numTrees
number of trees to use (-1 => number of training instances)
- useJackknife
whether to use jackknife based variance estimates
- biasLearner
learner to model bias (absolute residual)
- leafLearner
learner to use at the leaves of the trees
- subsetStrategy
for random feature selection at each split (auto => all fetures for regression, sqrt for classification)
- minLeafInstances
minimum number of instances per leave in each tree
- maxDepth
maximum depth of each tree in the forest (default: unlimited)
- uncertaintyCalibration
whether to empirically recalibrate the predicted uncertainties (default: false)
- randomizePivotLocation
whether to generate splits randomly between the data points (default: false)
- randomlyRotateFeatures
whether to randomly rotate real features for each tree in the forest (default: false)
- rng
random number generator to use for stochastic functionality
- Alphabetic
- By Inheritance
- RandomForest
- Product
- Equals
- Learner
- Serializable
- AnyRef
- Any
- Hide All
- Show All
- Public
- Protected
Instance Constructors
- new RandomForest(numTrees: Int = -1, useJackknife: Boolean = true, biasLearner: Option[Learner] = None, leafLearner: Option[Learner] = None, subsetStrategy: Any = "auto", minLeafInstances: Int = 1, maxDepth: Int = Integer.MAX_VALUE, uncertaintyCalibration: Boolean = true, randomizePivotLocation: Boolean = false, randomlyRotateFeatures: Boolean = false, rng: Random = Random)
- numTrees
number of trees to use (-1 => number of training instances)
- useJackknife
whether to use jackknife based variance estimates
- biasLearner
learner to model bias (absolute residual)
- leafLearner
learner to use at the leaves of the trees
- subsetStrategy
for random feature selection at each split (auto => all fetures for regression, sqrt for classification)
- minLeafInstances
minimum number of instances per leave in each tree
- maxDepth
maximum depth of each tree in the forest (default: unlimited)
- uncertaintyCalibration
whether to empirically recalibrate the predicted uncertainties (default: false)
- randomizePivotLocation
whether to generate splits randomly between the data points (default: false)
- randomlyRotateFeatures
whether to randomly rotate real features for each tree in the forest (default: false)
- rng
random number generator to use for stochastic functionality
Value Members
- final def !=(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
- final def ##: Int
- Definition Classes
- AnyRef → Any
- final def ==(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
- final def asInstanceOf[T0]: T0
- Definition Classes
- Any
- val biasLearner: Option[Learner]
- def clone(): AnyRef
- Attributes
- protected[lang]
- Definition Classes
- AnyRef
- Annotations
- @throws(classOf[java.lang.CloneNotSupportedException]) @native()
- final def eq(arg0: AnyRef): Boolean
- Definition Classes
- AnyRef
- def finalize(): Unit
- Attributes
- protected[lang]
- Definition Classes
- AnyRef
- Annotations
- @throws(classOf[java.lang.Throwable])
- final def getClass(): Class[_ <: AnyRef]
- Definition Classes
- AnyRef → Any
- Annotations
- @native()
- final def isInstanceOf[T0]: Boolean
- Definition Classes
- Any
- val leafLearner: Option[Learner]
- val maxDepth: Int
- val minLeafInstances: Int
- final def ne(arg0: AnyRef): Boolean
- Definition Classes
- AnyRef
- final def notify(): Unit
- Definition Classes
- AnyRef
- Annotations
- @native()
- final def notifyAll(): Unit
- Definition Classes
- AnyRef
- Annotations
- @native()
- val numTrees: Int
- def productElementNames: Iterator[String]
- Definition Classes
- Product
- val randomizePivotLocation: Boolean
- val randomlyRotateFeatures: Boolean
- val rng: Random
- val subsetStrategy: Any
- final def synchronized[T0](arg0: => T0): T0
- Definition Classes
- AnyRef
- def train(trainingData: Seq[(Vector[Any], Any)], weights: Option[Seq[Double]]): TrainingResult
Train a random forest model
Train a random forest model
If the training labels are Doubles, this is a regression forest; otherwise, a classification forest. Options like the number of trees are set via setHyper
- trainingData
to train on
- weights
for the training rows, if applicable
- returns
training result containing a model
- Definition Classes
- RandomForest → Learner
- def train(trainingData: Seq[(Vector[Any], Any, Double)]): TrainingResult
Train a model with weights
Train a model with weights
- trainingData
with weights in the form (features, label, weight)
- returns
training result containing a model
- Definition Classes
- Learner
- val uncertaintyCalibration: Boolean
- val useJackknife: Boolean
- final def wait(): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws(classOf[java.lang.InterruptedException])
- final def wait(arg0: Long, arg1: Int): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws(classOf[java.lang.InterruptedException])
- final def wait(arg0: Long): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws(classOf[java.lang.InterruptedException]) @native()