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
,Serializable
Gradient boosted DART tree implementation.
- See Also:
-
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
FieldsFields inherited from class biz.k11i.xgboost.gbm.GBTree
_groupTrees, mparam
-
Constructor Summary
Constructors -
Method Summary
Modifier and TypeMethodDescriptionvoid
loadModel
(ModelReader reader, boolean with_pbuffer) Loads model from stream.(package private) float
Methods inherited from class biz.k11i.xgboost.gbm.GBTree
predict, predictLeaf, predictSingle, predPath
Methods inherited from class biz.k11i.xgboost.gbm.GBBase
setNumClass
-
Field Details
-
weightDrop
private float[] weightDrop
-
-
Constructor Details
-
Dart
Dart()
-
-
Method Details
-
loadModel
Description copied from interface:GradBooster
Loads model from stream.- Specified by:
loadModel
in interfaceGradBooster
- Overrides:
loadModel
in classGBTree
- Parameters:
reader
- input streamwith_pbuffer
- whether the incoming data contains pbuffer- Throws:
IOException
- If an I/O error occurs
-
pred
-