Package biz.k11i.xgboost.gbm
Interface GradBooster
- All Superinterfaces:
Serializable
Interface of gradient boosting model.
-
Nested Class Summary
Nested Classes -
Method Summary
Modifier and TypeMethodDescriptionvoid
loadModel
(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.float
predictSingle
(FVec feat, int ntree_limit) Generates a prediction for given feature vector.void
setNumClass
(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
-