case class RegressionTreeLearner(numFeatures: Int = -1, maxDepth: Int = 30, minLeafInstances: Int = 1, leafLearner: Option[Learner] = None, splitter: Splitter[Double] = RegressionSplitter(), rng: Random = Random) extends Learner with Product with Serializable
Learner for regression trees
Created by maxhutch on 11/28/16.
- numFeatures
to randomly select from at each split (default: all)
- maxDepth
to grow the tree to
- minLeafInstances
minimum number of training instances per leaf
- leafLearner
learner to train the leaves with
- splitter
to determine the best split of the node data
- rng
random number generator, for reproducibility
- Alphabetic
- By Inheritance
- RegressionTreeLearner
- Product
- Equals
- Learner
- Serializable
- AnyRef
- Any
- Hide All
- Show All
- Public
- Protected
Instance Constructors
- new RegressionTreeLearner(numFeatures: Int = -1, maxDepth: Int = 30, minLeafInstances: Int = 1, leafLearner: Option[Learner] = None, splitter: Splitter[Double] = RegressionSplitter(), rng: Random = Random)
- numFeatures
to randomly select from at each split (default: all)
- maxDepth
to grow the tree to
- minLeafInstances
minimum number of training instances per leaf
- leafLearner
learner to train the leaves with
- splitter
to determine the best split of the node data
- rng
random number generator, for reproducibility
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
- 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 numFeatures: Int
- def productElementNames: Iterator[String]
- Definition Classes
- Product
- val rng: Random
- val splitter: Splitter[Double]
- final def synchronized[T0](arg0: => T0): T0
- Definition Classes
- AnyRef
- def train(trainingData: Seq[(Vector[Any], Any)], weights: Option[Seq[Double]] = None): RegressionTreeTrainingResult
Train the tree by recursively partitioning (splitting) the training data on a single feature
Train the tree by recursively partitioning (splitting) the training data on a single feature
- trainingData
to train on
- weights
for the training rows, if applicable
- returns
a RegressionTree
- Definition Classes
- RegressionTreeLearner → 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
- 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()