package regression
Ordering
- Alphabetic
Visibility
- Public
- Protected
Type Members
- class RegressionTrainingLeaf extends TrainingNode[AnyVal, Double]
Training leaf node for regression trees Created by maxhutch on 3/8/17.
- class RegressionTrainingNode extends TrainingNode[AnyVal, Double]
Created by maxhutch on 1/12/17.
- class RegressionTree extends Model[RegressionTreeResult]
Container holding a model node, encoders, and the feature influences
- 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
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
- class RegressionTreeResult extends PredictionResult[Double]
Simple wrapper around a sequence of predictions
- class RegressionTreeTrainingResult extends TrainingResult
Value Members
- object RegressionTrainingNode extends Serializable
Companion object to hold helper functions