classification

Type Members

1. trait BinaryClassificationModel extends MultinomialClassificationModel

   Trait for binary classifiers.

2. trait ClassificationModel extends AnyRef

   Trait for all classification models.

3. case class DecisionTreeClassifierModel(rootNode: Node, numFeatures: Int, numClasses: Int) extends MultinomialClassificationModel with DecisionTree with Serializable with Product

   Class for decision tree classification models.

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

   Class for a gradient boost classifier model.

5. case class LogisticRegressionModel(coefficients: Vector, intercept: Double, threshold: Option[Double] = scala.Some.apply[Double](0.5)) extends BinaryClassificationModel with Serializable with Product

   Class for binary logistic regression models.

6. trait MultinomialClassificationModel extends ClassificationModel

   Trait for classification models.

7. case class OneVsRestModel(classifiers: Array[BinaryClassificationModel]) extends Product with Serializable

   Class for multinomial one vs rest models.

8. case class RandomForestClassifierModel(trees: Seq[DecisionTreeClassifierModel], treeWeights: Seq[Double], numFeatures: Int, numClasses: Int) extends MultinomialClassificationModel with TreeEnsemble with Serializable with Product

   Class for random forest classification models.

9. case class SupportVectorMachineModel(coefficients: Vector, intercept: Double, threshold: Option[Double] = ...) extends BinaryClassificationModel with Serializable with Product

   Class for support vector machine models.