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
water.Key<hex.DataInfo>	_dataInfoKey
java.lang.String	_featureMap
XGBoostModel.XGBoostParameters	parameters

Constructor Summary

Constructors

Constructor and Description
XGBoostModelInfo(XGBoostModel.XGBoostParameters origParams) Main constructor

Method Summary

Modifier and Type	Method and Description
ml.dmlc.xgboost4j.java.Booster	deserializeBooster()
java.lang.String	getFeatureMap()
int	hashCode()
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

_featureMap

public java.lang.String _featureMap

_boosterBytes

public byte[] _boosterBytes

_dataInfoKey

public water.Key<hex.DataInfo> _dataInfoKey

parameters

public XGBoostModel.XGBoostParameters parameters

Constructor Detail

XGBoostModelInfo

public XGBoostModelInfo(XGBoostModel.XGBoostParameters origParams)

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()