Class GBTree

java.lang.Object
biz.k11i.xgboost.gbm.GBBase
biz.k11i.xgboost.gbm.GBTree
All Implemented Interfaces:
GradBooster, Serializable
Direct Known Subclasses:
Dart

public class GBTree extends GBBase
Gradient boosted tree implementation.
See Also:
  • Field Details

  • Constructor Details

    • GBTree

      GBTree()
  • Method Details

    • loadModel

      public void loadModel(ModelReader reader, boolean with_pbuffer) throws IOException
      Description copied from interface: GradBooster
      Loads model from stream.
      Parameters:
      reader - input stream
      with_pbuffer - whether the incoming data contains pbuffer
      Throws:
      IOException - If an I/O error occurs
    • predict

      public float[] predict(FVec feat, int ntree_limit)
      Description copied from interface: GradBooster
      Generates predictions for given feature vector.
      Parameters:
      feat - feature vector
      ntree_limit - limit the number of trees used in prediction
      Returns:
      prediction result
    • predictSingle

      public float predictSingle(FVec feat, int ntree_limit)
      Description copied from interface: GradBooster
      Generates a prediction for given feature vector.

      This method only works when the model outputs single value.

      Parameters:
      feat - feature vector
      ntree_limit - limit the number of trees used in prediction
      Returns:
      prediction result
    • pred

      float pred(FVec feat, int bst_group, int root_index, int ntree_limit)
    • predictLeaf

      public int[] predictLeaf(FVec feat, int ntree_limit)
      Description copied from interface: GradBooster
      Predicts the leaf index of each tree. This is only valid in gbtree predictor.
      Parameters:
      feat - feature vector
      ntree_limit - limit the number of trees used in prediction
      Returns:
      predicted leaf indexes
    • predPath

      int[] predPath(FVec feat, int root_index, int ntree_limit)