Class/Object

ml.combust.mleap.core.classification

GBTClassifierModel

Related Docs: object GBTClassifierModel | package classification

Permalink

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
Visibility
  1. Public
  2. All

Instance Constructors

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

    Permalink

    trees

    trees in the gradient boost model

    treeWeights

    weights of each tree

    numFeatures

    number of features

Value Members

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

    Permalink
    Definition Classes
    AnyRef → Any
  2. final def ##(): Int

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

    Permalink
    Definition Classes
    AnyRef → Any
  4. def apply(features: Vector): Double

    Permalink

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

    features

    feature vector

    returns

    prediction

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

    Permalink
    Definition Classes
    Any
  6. def binaryProbabilityToPrediction(probability: Double): Double

    Permalink
    Definition Classes
    BinaryClassificationModel
  7. def clone(): AnyRef

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

    Permalink
    Definition Classes
    AnyRef
  9. def finalize(): Unit

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

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

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

    Permalink
    Definition Classes
    AnyRef
  13. final def notify(): Unit

    Permalink
    Definition Classes
    AnyRef
  14. final def notifyAll(): Unit

    Permalink
    Definition Classes
    AnyRef
  15. val numClasses: Int

    Permalink

    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
  16. val numFeatures: Int

    Permalink

    number of features

  17. def numTrees: Int

    Permalink

    Number of trees in the ensemble

    Number of trees in the ensemble

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

    Permalink

    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
  19. def predictBinaryProbability(features: Vector): Double

    Permalink

    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
  20. def predictBinaryWithProbability(features: Vector): (Double, Double)

    Permalink

    Predict class and probability.

    Predict class and probability.

    features

    features to predict

    returns

    (prediction, probability)

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

    Permalink
  22. def predictRaw(features: Vector): Vector

    Permalink
  23. def predictWithProbability(features: Vector): (Double, Double)

    Permalink
  24. def probabilityToPrediction(probability: Vector): Double

    Permalink
  25. def rawToPrediction(raw: Vector): Double

    Permalink
  26. def rawToProbability(raw: Vector): Vector

    Permalink
  27. def rawToProbabilityInPlace(raw: Vector): Vector

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

    Permalink
    Definition Classes
    AnyRef
  29. val threshold: Option[Double]

    Permalink

    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
  30. lazy val thresholds: Option[Array[Double]]

    Permalink
  31. val treeWeights: Seq[Double]

    Permalink

    weights of each tree

    weights of each tree

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

    Permalink

    trees in the gradient boost model

    trees in the gradient boost model

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

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

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

    Permalink
    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