Package biz.k11i.xgboost.gbm
Class Dart
- java.lang.Object
-
- biz.k11i.xgboost.gbm.GBBase
-
- biz.k11i.xgboost.gbm.GBTree
-
- biz.k11i.xgboost.gbm.Dart
-
- All Implemented Interfaces:
GradBooster,java.io.Serializable
public class Dart extends GBTree
Gradient boosted DART tree implementation.- See Also:
- Serialized Form
-
-
Nested Class Summary
-
Nested classes/interfaces inherited from class biz.k11i.xgboost.gbm.GBTree
GBTree.ModelParam
-
Nested classes/interfaces inherited from interface biz.k11i.xgboost.gbm.GradBooster
GradBooster.Factory
-
-
Field Summary
Fields Modifier and Type Field Description private float[]weightDrop-
Fields inherited from class biz.k11i.xgboost.gbm.GBTree
_groupTrees, mparam
-
Fields inherited from class biz.k11i.xgboost.gbm.GBBase
num_class, num_feature, num_output_group
-
-
Constructor Summary
Constructors Constructor Description Dart()
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description voidloadModel(PredictorConfiguration config, ModelReader reader, boolean with_pbuffer)Loads model from stream.(package private) floatpred(FVec feat, int bst_group, int root_index, int ntree_limit, float base_score)floatweight(int tidx)-
Methods inherited from class biz.k11i.xgboost.gbm.GBTree
getGroupedTrees, predict, predictLeaf, predictLeafPath, predictSingle
-
Methods inherited from class biz.k11i.xgboost.gbm.GBBase
setNumClass, setNumFeature
-
-
-
-
Method Detail
-
loadModel
public void loadModel(PredictorConfiguration config, ModelReader reader, boolean with_pbuffer) throws java.io.IOException
Description copied from interface:GradBoosterLoads model from stream.- Specified by:
loadModelin interfaceGradBooster- Overrides:
loadModelin classGBTree- Parameters:
config- predictor configurationreader- input streamwith_pbuffer- whether the incoming data contains pbuffer- Throws:
java.io.IOException- If an I/O error occurs
-
pred
float pred(FVec feat, int bst_group, int root_index, int ntree_limit, float base_score)
-
weight
public float weight(int tidx)
-
-