ml.combust.mleap.core.classification

GBTClassifierModel

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.

trees

trees in the gradient boost model

treeWeights

weights of each tree

numFeatures

number of features

Linear Supertypes
Product, Equals, Serializable, Serializable, TreeEnsemble, BinaryClassificationModel, MultinomialClassificationModel, ClassificationModel, AnyRef, Any
Ordering
  1. Alphabetic
  2. By inheritance
Inherited
  1. GBTClassifierModel
  2. Product
  3. Equals
  4. Serializable
  5. Serializable
  6. TreeEnsemble
  7. BinaryClassificationModel
  8. MultinomialClassificationModel
  9. ClassificationModel
  10. AnyRef
  11. Any
  1. Hide All
  2. Show all
Learn more about member selection
Visibility
  1. Public
  2. All

Instance Constructors

  1. new GBTClassifierModel(trees: Seq[DecisionTreeRegressionModel], treeWeights: Seq[Double], numFeatures: Int)

    trees

    trees in the gradient boost model

    treeWeights

    weights of each tree

    numFeatures

    number of features

Value Members

  1. final def !=(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  2. final def !=(arg0: Any): Boolean

    Definition Classes
    Any
  3. final def ##(): Int

    Definition Classes
    AnyRef → Any
  4. final def ==(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  5. final def ==(arg0: Any): Boolean

    Definition Classes
    Any
  6. def apply(features: Vector): Double

    Alias for ml.combust.mleap.core.classification.ClassificationModel#predict.

    features

    feature vector

    returns

    prediction

    Definition Classes
    ClassificationModel
  7. final def asInstanceOf[T0]: T0

    Definition Classes
    Any
  8. def binaryProbabilityToPrediction(probability: Double): Double

    Definition Classes
    BinaryClassificationModel
  9. def clone(): AnyRef

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  10. final def eq(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  11. def finalize(): Unit

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  12. final def getClass(): Class[_]

    Definition Classes
    AnyRef → Any
  13. final def isInstanceOf[T0]: Boolean

    Definition Classes
    Any
  14. final def ne(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  15. final def notify(): Unit

    Definition Classes
    AnyRef
  16. final def notifyAll(): Unit

    Definition Classes
    AnyRef
  17. val numClasses: Int

    Number of classes this model predicts.

    Number of classes this model predicts.

    2 indicates this is a binary classification model. Greater than 2 indicates a multinomial classifier.

    Definition Classes
    BinaryClassificationModelMultinomialClassificationModel
  18. val numFeatures: Int

    number of features

  19. def numTrees: Int

    Number of trees in the ensemble

    Number of trees in the ensemble

    Definition Classes
    TreeEnsemble
  20. def predict(features: Vector): Double

    Predict the class taking into account threshold.

    Predict the class taking into account threshold.

    features

    features for prediction

    returns

    prediction with threshold

    Definition Classes
    BinaryClassificationModelMultinomialClassificationModelClassificationModel
  21. def predictBinaryProbability(features: Vector): Double

    Predict the class without taking into account threshold.

    Predict the class without taking into account threshold.

    features

    features for prediction

    returns

    probability that prediction is the predictable class

    Definition Classes
    GBTClassifierModelBinaryClassificationModel
  22. def predictBinaryWithProbability(features: Vector): (Double, Double)

    Predict class and probability.

    Predict class and probability.

    features

    features to predict

    returns

    (prediction, probability)

    Definition Classes
    BinaryClassificationModel
  23. def predictProbabilities(features: Vector): Vector

  24. def predictRaw(features: Vector): Vector

  25. def predictWithProbability(features: Vector): (Double, Double)

  26. def probabilityToPrediction(probability: Vector): Double

  27. def rawToPrediction(raw: Vector): Double

  28. def rawToProbability(raw: Vector): Vector

  29. def rawToProbabilityInPlace(raw: Vector): Vector

  30. final def synchronized[T0](arg0: ⇒ T0): T0

    Definition Classes
    AnyRef
  31. val threshold: Option[Double]

    Threshold for binary classifiers.

    Threshold for binary classifiers.

    If the prediction probability is over this value, then the prediction is pegged to 1.0. Otherwise the prediction is pegged to 0.0.

    Definition Classes
    GBTClassifierModelBinaryClassificationModel
  32. lazy val thresholds: Option[Array[Double]]

  33. val treeWeights: Seq[Double]

    weights of each tree

    weights of each tree

    Definition Classes
    GBTClassifierModelTreeEnsemble
  34. val trees: Seq[DecisionTreeRegressionModel]

    trees in the gradient boost model

    trees in the gradient boost model

    Definition Classes
    GBTClassifierModelTreeEnsemble
  35. final def wait(): Unit

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  36. final def wait(arg0: Long, arg1: Int): Unit

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  37. final def wait(arg0: Long): Unit

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

Inherited from Product

Inherited from Equals

Inherited from Serializable

Inherited from Serializable

Inherited from TreeEnsemble

Inherited from BinaryClassificationModel

Inherited from ClassificationModel

Inherited from AnyRef

Inherited from Any

Ungrouped