regression

Type Members

1. case class AFTSurvivalRegressionModel(coefficients: Vector, intercept: Double, quantileProbabilities: Array[Double], scale: Double) extends Model with Product with Serializable

   

   Created by hollinwilkins on 12/28/16.

   Created by hollinwilkins on 12/28/16.

   Annotations

   @SparkCode()

2. case class DecisionTreeRegressionModel(rootNode: Node, numFeatures: Int) extends DecisionTree with Model 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

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

   

   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

4. case class GeneralizedLinearRegressionModel(coefficients: Vector, intercept: Double, fal: FamilyAndLink) extends Model with Product with Serializable

   

5. case class IsotonicRegressionModel(boundaries: Array[Double], predictions: Seq[Double], isotonic: Boolean, featureIndex: Option[Int]) extends Model with Product with Serializable

   

   Created by hollinwilkins on 12/27/16.

   Created by hollinwilkins on 12/27/16.

   Annotations

   @SparkCode()

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

   

   Class for linear regression model.

   Class for linear regression model.

   coefficients

   coefficients for linear regression

   intercept

   intercept for regression

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

   

   Class for random forest regression.

   Class for random forest regression.

   trees

   trees in the random forest

   numFeatures

   number of features needed for prediction