ml.combust.mleap.core

regression

package regression

Visibility
  1. Public
  2. All

Type Members

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

    Class for a decision tree regression model.

  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.

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

    Class for linear regression model.

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

    Class for random forest regression.

Value Members

  1. object GBTRegressionModel extends Serializable

    Companion object for constructing GBTRegressionModel.

  2. object RandomForestRegressionModel extends Serializable

    Companion object for constructing RandomForestRegressionModel.

Ungrouped