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, int nClasses) Main constructor

Method Summary

Modifier and Type	Method and Description
ml.dmlc.xgboost4j.java.Booster	booster()
XGBoostModel.XGBoostParameters	get_params()
ml.dmlc.xgboost4j.java.Booster	getBooster()
java.lang.String	getFeatureMap()
int	hashCode()
void	nativeToJava()
void	nukeBackend()
void	setBooster(ml.dmlc.xgboost4j.java.Booster _booster)
void	setFeatureMap(java.lang.String featureMap)
long	size()
java.lang.String	toString() Print a summary table

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, 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,
                        int nClasses)

Main constructor

Parameters:

origParams - Model parameters

nClasses - number of classes (1 for regression, 0 for autoencoder)

Method Detail

getFeatureMap

public java.lang.String getFeatureMap()

setFeatureMap

public void setFeatureMap(java.lang.String featureMap)

getBooster

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

setBooster

public void setBooster(ml.dmlc.xgboost4j.java.Booster _booster)

booster

public final ml.dmlc.xgboost4j.java.Booster booster()

nukeBackend

public void nukeBackend()

nativeToJava

public void nativeToJava()

hashCode

public int hashCode()

Overrides:

hashCode in class java.lang.Object

size

public long size()

get_params

public final XGBoostModel.XGBoostParameters get_params()

toString

public java.lang.String toString()

Print a summary table

Overrides:

toString in class java.lang.Object

Returns:

String containing ASCII version of summary table