Package biz.k11i.xgboost.gbm
Class Dart
- java.lang.Object
-
- biz.k11i.xgboost.gbm.GBTree
-
- biz.k11i.xgboost.gbm.Dart
-
- All Implemented Interfaces:
GradBooster
,Serializable
public class Dart extends GBTree
Gradient boosted DART tree implementation.- See Also:
- Serialized Form
-
-
Nested Class Summary
-
Nested classes/interfaces inherited from interface biz.k11i.xgboost.gbm.GradBooster
GradBooster.Factory
-
-
Field Summary
Fields Modifier and Type Field Description protected int
num_class
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description void
loadModel(ModelReader reader, boolean with_pbuffer)
Loads model from stream.void
setNumClass(int num_class)
-
Methods inherited from class biz.k11i.xgboost.gbm.GBTree
predict, predictLeaf, predictSingle
-
-
-
-
Method Detail
-
loadModel
public void loadModel(ModelReader reader, boolean with_pbuffer) throws IOException
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
-
setNumClass
public void setNumClass(int num_class)
- Specified by:
setNumClass
in interfaceGradBooster
-
-