org.jpmml.xgboost

Class GBTree

java.lang.Object

org.jpmml.xgboost.GradientBooster

org.jpmml.xgboost.GBTree

All Implemented Interfaces:

BinaryLoadable, JSONLoadable, UBJSONLoadable

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, org.jpmml.converter.Schema schema)
String	getAlgorithmName()
BitSet	getSplitCategories(int splitIndex)
Set<Integer>	getSplitType(int splitIndex)
boolean	hasCategoricalSplits()
void	loadBinary(XGBoostDataInput input)
void	loadJSON(com.google.gson.JsonObject gradientBooster)
void	loadUBJSON(com.devsmart.ubjson.UBObject gradientBooster)
int	num_parallel_tree()
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)

loadUBJSON

public void loadUBJSON(com.devsmart.ubjson.UBObject gradientBooster)

hasCategoricalSplits

public boolean hasCategoricalSplits()

getSplitType

public Set<Integer> getSplitType(int splitIndex)

getSplitCategories

public BitSet getSplitCategories(int splitIndex)

encodeMiningModel

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

num_trees

public int num_trees()

num_parallel_tree

public int num_parallel_tree()

trees

public RegTree[] trees()

tree_weights

public float[] tree_weights()