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 GradBooster

GradBooster.Factory

Field Summary

Fields

Modifier and Type	Field	Description
protected int	num_class
protected int	num_feature
protected int	num_output_group

Method Summary

Modifier and Type	Method	Description
void	loadModel(PredictorConfiguration config, ModelReader reader, boolean with_pbuffer)	Loads model from stream.
void	setNumClass(int numClass)
void	setNumFeature(int numFeature)
float	weight(int tidx)

Methods inherited from class GBTree

getGroupedTrees, predict, predictLeaf, predictLeafPath, predictSingle

Methods inherited from class Object

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

Field Details

num_class

protected int num_class

num_feature

protected int num_feature

num_output_group

protected int num_output_group

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

weight

public float weight(int tidx)

setNumClass

public void setNumClass(int numClass)

Specified by:

setNumClass in interface GradBooster

setNumFeature

public void setNumFeature(int numFeature)

Specified by:

setNumFeature in interface GradBooster