org.jpmml.xgboost

Class GBTree

java.lang.Object

org.jpmml.xgboost.GradientBooster

org.jpmml.xgboost.GBTree

All Implemented Interfaces:

BinaryLoadable, JSONLoadable

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, Integer ntreeLimit, boolean numeric, org.jpmml.converter.Schema schema)
String	getAlgorithmName()
void	loadBinary(XGBoostDataInput input)
void	loadJSON(com.google.gson.JsonObject gradientBooster)
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 String getAlgorithmName()

Specified by:

getAlgorithmName in class GradientBooster

loadBinary

public void loadBinary(XGBoostDataInput input)
                throws IOException

Throws:

IOException

loadJSON

public void loadJSON(com.google.gson.JsonObject gradientBooster)

encodeMiningModel

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

num_trees

public int num_trees()

trees

public RegTree[] trees()

tree_weights

public float[] tree_weights()