biz.k11i.xgboost.gbm

Class Dart

java.lang.Object

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 interface biz.k11i.xgboost.gbm.GradBooster

GradBooster.Factory

Field Summary

Fields

Modifier and Type	Field and Description
protected int	num_class

Method Summary

Methods

Modifier and Type	Method and Description
void	loadModel(PredictorConfiguration config, ModelReader reader, boolean with_pbuffer) Loads model from stream.
void	setNumClass(int num_class)

Methods inherited from class biz.k11i.xgboost.gbm.GBTree

getGroupedTrees, predict, predictLeaf, predictSingle

Methods inherited from class java.lang.Object

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

Field Detail

num_class

protected int num_class

Method Detail

loadModel

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

Description copied from interface: GradBooster

Loads model from stream.

Specified by:

loadModel in interface GradBooster

Overrides:

loadModel in class GBTree

reader - input stream

with_pbuffer - whether the incoming data contains pbuffer

Throws:

java.io.IOException - If an I/O error occurs

setNumClass

public void setNumClass(int num_class)

Specified by:

setNumClass in interface GradBooster