tree

Type Members

1. final case class CategoricalSplit(featureIndex: Int, numCategories: Int, categories: Array[Double], isLeft: Boolean) extends Split with Product with Serializable

   Class for splitting on a categorical feature.

2. final case class ContinuousSplit(featureIndex: Int, threshold: Double) extends Split with Product with Serializable

   Class for splitting on a continuous feature.

3. trait DecisionTree extends Serializable

   Trait for a decision tree.

4. final case class InternalNode(left: Node, right: Node, split: Split) extends Node with Product with Serializable

   Trait for internal node in a decision tree.

5. final case class LeafNode(prediction: Double, impurities: Option[Vector] = scala.None) extends Node with Product with Serializable

   Trait for a leaf node in a decision tree.

6. sealed trait Node extends Serializable

   Trait for a node in a decision tree.

7. sealed trait Split extends Serializable

   Trait for a split logic.

8. trait TreeEnsemble extends AnyRef

   Trait for an ensemble of decision trees.