hex.tree.xgboost

Class XGBoostModelInfo

java.lang.Object

water.Iced

hex.tree.xgboost.XGBoostModelInfo

All Implemented Interfaces:

java.io.Externalizable, java.io.Serializable, java.lang.Cloneable, water.Freezable

public final class XGBoostModelInfo
extends water.Iced

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
byte[]	_boosterBytes
java.lang.String	_featureMap
XGBoostModel.XGBoostParameters	_parameters

Constructor Summary

Constructors

Constructor and Description
XGBoostModelInfo(XGBoostModel.XGBoostParameters origParams, hex.DataInfo dinfo) Main constructor

Method Summary

Modifier and Type	Method and Description
byte[]	auxNodeWeightBytes()
AuxNodeWeights	auxNodeWeights()
hex.DataInfo	dataInfo()
water.Key<AuxNodeWeights>	getAuxNodeWeightsKey()
water.Key<hex.DataInfo>	getDataInfoKey()
java.lang.String	getFeatureMap()
int	hashCode()
hex.DataInfo	scoringInfo(boolean isTrain)
void	setFeatureMap(java.lang.String featureMap)
long	size()
void	updateBoosterBytes(byte[] boosterBytes)

Methods inherited from class water.Iced

asBytes, clone, copyOver, frozenType, read, readExternal, readJSON, reloadFromBytes, toJsonBytes, toJsonString, write, writeExternal, writeJSON

Methods inherited from class java.lang.Object

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

Field Detail

_parameters

public final XGBoostModel.XGBoostParameters _parameters

_featureMap

public java.lang.String _featureMap

_boosterBytes

public byte[] _boosterBytes

Constructor Detail

XGBoostModelInfo

public XGBoostModelInfo(XGBoostModel.XGBoostParameters origParams,
                        hex.DataInfo dinfo)

Main constructor

Parameters:

origParams - Model parameters

Method Detail

getFeatureMap

public java.lang.String getFeatureMap()

setFeatureMap

public void setFeatureMap(java.lang.String featureMap)

updateBoosterBytes

public void updateBoosterBytes(byte[] boosterBytes)

hashCode

public int hashCode()

Overrides:

hashCode in class java.lang.Object

size

public long size()

getDataInfoKey

public water.Key<hex.DataInfo> getDataInfoKey()

dataInfo

public hex.DataInfo dataInfo()

scoringInfo

public hex.DataInfo scoringInfo(boolean isTrain)

getAuxNodeWeightsKey

public water.Key<AuxNodeWeights> getAuxNodeWeightsKey()

auxNodeWeights

public AuxNodeWeights auxNodeWeights()

auxNodeWeightBytes

public byte[] auxNodeWeightBytes()