Package biz.k11i.xgboost.gbm
Interface GradBooster
- All Superinterfaces:
Serializable
Interface of gradient boosting model.
-
Nested Class Summary
Nested Classes -
Method Summary
Modifier and TypeMethodDescriptionvoidloadModel(ModelReader reader, boolean with_pbuffer) Loads model from stream.float[]Generates predictions for given feature vector.int[]predictLeaf(FVec feat, int ntree_limit) Predicts the leaf index of each tree.floatpredictSingle(FVec feat, int ntree_limit) Generates a prediction for given feature vector.voidsetNumClass(int num_class)
-
Method Details
-
setNumClass
void setNumClass(int num_class) -
loadModel
Loads model from stream.- Parameters:
reader- input streamwith_pbuffer- whether the incoming data contains pbuffer- Throws:
IOException- If an I/O error occurs
-
predict
Generates predictions for given feature vector.- Parameters:
feat- feature vectorntree_limit- limit the number of trees used in prediction- Returns:
- prediction result
-
predictSingle
Generates a prediction for given feature vector.This method only works when the model outputs single value.
- Parameters:
feat- feature vectorntree_limit- limit the number of trees used in prediction- Returns:
- prediction result
-
predictLeaf
Predicts the leaf index of each tree. This is only valid in gbtree predictor.- Parameters:
feat- feature vectorntree_limit- limit the number of trees used in prediction- Returns:
- predicted leaf indexes
-