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) Used when executing from a local model

Method Summary

Modifier and Type	Method and Description
void	close()
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)

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

updateBooster

public byte[] updateBooster()

Specified by:

updateBooster in interface XGBoostExecutor

close

public void close()

Specified by:

close in interface java.lang.AutoCloseable