regression

Type Members

1. case class DecisionTreeRegressionModel(rootNode: Node, numFeatures: Int) extends DecisionTree with Product with Serializable

   

   Class for a decision tree regression model.

   Class for a decision tree regression model.

   rootNode

   root decision tree node

   numFeatures

   number of features used in prediction

2. case class GBTRegressionModel(trees: Seq[DecisionTreeRegressionModel], treeWeights: Seq[Double], numFeatures: Int) extends TreeEnsemble with Serializable with Product

   

   Class for gradient boosted tree regression model.

   Class for gradient boosted tree regression model.

   trees

   trees in model

   treeWeights

   weight of each tree

   numFeatures

   number of features

3. case class LinearRegressionModel(coefficients: Vector, intercept: Double) extends Serializable with Product

   

   Class for linear regression model.

   Class for linear regression model.

   coefficients

   coefficients for linear regression

   intercept

   intercept for regression

4. case class RandomForestRegressionModel(trees: Seq[DecisionTreeRegressionModel], treeWeights: Seq[Double], numFeatures: Int) extends TreeEnsemble with Serializable with Product

   

   Class for random forest regression.

   Class for random forest regression.

   trees

   trees in the random forest

   numFeatures

   number of features needed for prediction