Packages

c

io.citrine.lolo.trees.regression

RegressionTreeLearner

case class RegressionTreeLearner(numFeatures: Int = -1, maxDepth: Int = 30, minLeafInstances: Int = 1, leafLearner: Option[Learner] = None) 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

leafLearner

learner to train the leaves with

Linear Supertypes
Product, Equals, Learner, Serializable, Serializable, AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. RegressionTreeLearner
  2. Product
  3. Equals
  4. Learner
  5. Serializable
  6. Serializable
  7. AnyRef
  8. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new RegressionTreeLearner(numFeatures: Int = -1, maxDepth: Int = 30, minLeafInstances: Int = 1, leafLearner: Option[Learner] = None)

    numFeatures

    to randomly select from at each split (default: all)

    maxDepth

    to grow the tree to

    leafLearner

    learner to train the leaves with

Value Members

  1. final def !=(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  2. final def ##(): Int
    Definition Classes
    AnyRef → Any
  3. final def ==(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  4. final def asInstanceOf[T0]: T0
    Definition Classes
    Any
  5. def clone(): AnyRef
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @native() @throws( ... )
  6. final def eq(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  7. def finalize(): Unit
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  8. final def getClass(): Class[_]
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  9. final def isInstanceOf[T0]: Boolean
    Definition Classes
    Any
  10. val leafLearner: Option[Learner]
  11. val maxDepth: Int
  12. val minLeafInstances: Int
  13. final def ne(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  14. final def notify(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  15. final def notifyAll(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  16. val numFeatures: Int
  17. final def synchronized[T0](arg0: ⇒ T0): T0
    Definition Classes
    AnyRef
  18. 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
    RegressionTreeLearnerLearner
  19. 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
  20. final def wait(): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  21. final def wait(arg0: Long, arg1: Int): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  22. final def wait(arg0: Long): Unit
    Definition Classes
    AnyRef
    Annotations
    @native() @throws( ... )

Inherited from Product

Inherited from Equals

Inherited from Learner

Inherited from Serializable

Inherited from Serializable

Inherited from AnyRef

Inherited from Any

Ungrouped