hex.tree.xgboost.exec

Class LocalXGBoostExecutor

java.lang.Object

hex.tree.xgboost.exec.LocalXGBoostExecutor

All Implemented Interfaces:

XGBoostExecutor, java.lang.AutoCloseable

public class LocalXGBoostExecutor
extends java.lang.Object
implements XGBoostExecutor

Field Summary

Fields

Modifier and Type	Field and Description
water.Key	modelKey

Constructor Summary

Constructors

Constructor and Description
LocalXGBoostExecutor(water.Key key, XGBoostExecReq.Init init) Used when executing from a remote model
LocalXGBoostExecutor(XGBoostModel model, water.fvec.Frame train, water.fvec.Frame valid) Used when executing from a local model

Method Summary

Modifier and Type	Method and Description
void	close()
EvalMetric	getEvalMetric() Retrieves the value of the evaluation metric both for training and validation dataset.
byte[]	setup()
void	update(int treeId)
byte[]	updateBooster()

Methods inherited from class java.lang.Object

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

Field Detail

modelKey

public final water.Key modelKey

Constructor Detail

LocalXGBoostExecutor

public LocalXGBoostExecutor(water.Key key,
                            XGBoostExecReq.Init init)

Used when executing from a remote model

LocalXGBoostExecutor

public LocalXGBoostExecutor(XGBoostModel model,
                            water.fvec.Frame train,
                            water.fvec.Frame valid)

Used when executing from a local model

Method Detail

setup

public byte[] setup()

Specified by:

setup in interface XGBoostExecutor

update

public void update(int treeId)

Specified by:

update in interface XGBoostExecutor

getEvalMetric

public EvalMetric getEvalMetric()

Description copied from interface: XGBoostExecutor

Retrieves the value of the evaluation metric both for training and validation dataset.

Specified by:

getEvalMetric in interface XGBoostExecutor

Returns:

instance of EvalMetric if "eval_metric" was defined, null otherwise

updateBooster

public byte[] updateBooster()

Specified by:

updateBooster in interface XGBoostExecutor

close

public void close()

Specified by:

close in interface java.lang.AutoCloseable