org.jpmml.xgboost

Class GBTree

java.lang.Object

org.jpmml.xgboost.GradientBooster

org.jpmml.xgboost.GBTree

All Implemented Interfaces:

Loadable

Direct Known Subclasses:

Dart

public class GBTree
extends GradientBooster

Constructor Summary

Constructors

Constructor and Description
GBTree()

Method Summary

Modifier and Type	Method and Description
org.dmg.pmml.mining.MiningModel	encodeMiningModel(ObjFunction obj, float base_score, java.lang.Integer ntreeLimit, org.jpmml.converter.Schema schema)
java.lang.String	getAlgorithmName()
void	load(XGBoostDataInput input)
int	num_trees()
float[]	tree_weights()
RegTree[]	trees()

Methods inherited from class java.lang.Object

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

Constructor Detail

GBTree

public GBTree()

Method Detail

getAlgorithmName

public java.lang.String getAlgorithmName()

Specified by:

getAlgorithmName in class GradientBooster

load

public void load(XGBoostDataInput input)
          throws java.io.IOException

Throws:

java.io.IOException

encodeMiningModel

public org.dmg.pmml.mining.MiningModel encodeMiningModel(ObjFunction obj,
                                                         float base_score,
                                                         java.lang.Integer ntreeLimit,
                                                         org.jpmml.converter.Schema schema)

num_trees

public int num_trees()

trees

public RegTree[] trees()

tree_weights

public float[] tree_weights()