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

public class Dart
extends GBTree

Gradient boosted DART tree implementation.

See Also:

• Serialized Form

Nested Class Summary

Nested classes/interfaces inherited from class GBTree

GBTree.ModelParam

Nested classes/interfaces inherited from interface GradBooster

GradBooster.Factory

Field Summary

Fields

Modifier and Type	Field	Description
private float[]	weightDrop

Fields inherited from class GBTree

_groupTrees, mparam

Fields inherited from class GBBase

num_class, num_feature, num_output_group

Constructor Summary

Constructors

Constructor	Description
Dart()

Method Summary

Modifier and Type	Method	Description
void	loadModel(PredictorConfiguration config, ModelReader reader, boolean with_pbuffer)	Loads model from stream.
(package private) float	pred(FVec feat, int bst_group, int root_index, int ntree_limit, float base_score)
float	weight(int tidx)

Methods inherited from class GBTree

getGroupedTrees, predict, predictLeaf, predictLeafPath, predictSingle

Methods inherited from class GBBase

setNumClass, setNumFeature

Methods inherited from class Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Details

weightDrop

private float[] weightDrop

Constructor Details

Dart

Dart()

Method Details

loadModel

public void loadModel(PredictorConfiguration config,
 ModelReader reader,
 boolean with_pbuffer)
               throws IOException

Description copied from interface: GradBooster

Loads model from stream.

Specified by:

loadModel in interface GradBooster

Overrides:

loadModel in class GBTree

Parameters:

config - predictor configuration

reader - input stream

with_pbuffer - whether the incoming data contains pbuffer

Throws:

IOException - If an I/O error occurs

pred

float pred(FVec feat,
 int bst_group,
 int root_index,
 int ntree_limit,
 float base_score)

Overrides:

pred in class GBTree

weight

public float weight(int tidx)