org.jpmml.xgboost

Class RegTree

java.lang.Object

org.jpmml.xgboost.RegTree

All Implemented Interfaces:

BinaryLoadable, JSONLoadable, UBJSONLoadable

public class RegTree
extends Object
implements BinaryLoadable, JSONLoadable, UBJSONLoadable

Constructor Summary

Constructors

Constructor and Description
RegTree()

Method Summary

Modifier and Type	Method and Description
org.dmg.pmml.tree.TreeModel	encodeTreeModel(org.jpmml.converter.PredicateManager predicateManager, org.jpmml.converter.Schema schema)
Float	getLeafValue()
BitSet	getSplitCategories(int splitIndex)
Set<Integer>	getSplitType(int splitIndex)
boolean	hasCategoricalSplits()
void	loadBinary(XGBoostDataInput input)
void	loadJSON(com.google.gson.JsonObject tree)
void	loadUBJSON(com.devsmart.ubjson.UBObject tree)

Methods inherited from class java.lang.Object

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

Constructor Detail

RegTree

public RegTree()

Method Detail

loadBinary

public void loadBinary(XGBoostDataInput input)
                throws IOException

Specified by:

loadBinary in interface BinaryLoadable

Throws:

IOException

loadJSON

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

Specified by:

loadJSON in interface JSONLoadable

loadUBJSON

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

Specified by:

loadUBJSON in interface UBJSONLoadable

getLeafValue

public Float getLeafValue()

hasCategoricalSplits

public boolean hasCategoricalSplits()

getSplitType

public Set<Integer> getSplitType(int splitIndex)

getSplitCategories

public BitSet getSplitCategories(int splitIndex)

encodeTreeModel

public org.dmg.pmml.tree.TreeModel encodeTreeModel(org.jpmml.converter.PredicateManager predicateManager,
                                                   org.jpmml.converter.Schema schema)