ml.dmlc.xgboost4j.java

Class XGBoostModelInfo

java.lang.Object

water.Iced

ml.dmlc.xgboost4j.java.XGBoostModelInfo

All Implemented Interfaces:

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

public final class XGBoostModelInfo
extends water.Iced

This class contains the state of the Deep Learning model This will be shared: one per node

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
hex.DataInfo	dataInfo()
ml.dmlc.xgboost4j.java.Booster	deserializeBooster()
water.Key<hex.DataInfo>	getDataInfoKey()
java.lang.String	getFeatureMap()
int	hashCode()
hex.DataInfo	scoringInfo(boolean isTrain)
void	setBoosterBytes(byte[] boosterBytes)
void	setFeatureMap(java.lang.String featureMap)
long	size()

Methods inherited from class water.Iced

asBytes, clone, copyOver, frozenType, read, readExternal, readJSON, reloadFromBytes, 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)

setBoosterBytes

public void setBoosterBytes(byte[] boosterBytes)

deserializeBooster

public ml.dmlc.xgboost4j.java.Booster deserializeBooster()

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)