Class GBLinear

java.lang.Object
biz.k11i.xgboost.gbm.GBBase
biz.k11i.xgboost.gbm.GBLinear
All Implemented Interfaces:
GradBooster, Serializable

public class GBLinear extends GBBase
Linear booster implementation
See Also:
  • Field Details

  • Constructor Details

    • GBLinear

      GBLinear()
  • Method Details

    • loadModel

      public void loadModel(ModelReader reader, boolean ignored_with_pbuffer) throws IOException
      Description copied from interface: GradBooster
      Loads model from stream.
      Parameters:
      reader - input stream
      ignored_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 gid)
    • 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
    • weight

      float weight(int fid, int gid)
    • bias

      float bias(int gid)