ai.h2o.xgboost4j.java

Class BoosterWrapper

java.lang.Object

ai.h2o.xgboost4j.java.BoosterWrapper

public class BoosterWrapper
extends java.lang.Object

Wrapper to expose package private methods

Constructor Summary

Constructors

Constructor and Description
BoosterWrapper(byte[] checkpointBoosterBytes, java.util.Map<java.lang.String,java.lang.Object> params, ai.h2o.xgboost4j.java.DMatrix train)

Method Summary

Modifier and Type	Method and Description
void	dispose()
ai.h2o.xgboost4j.java.Booster	getBooster()
void	saveRabitCheckpoint()
byte[]	toByteArray()
void	update(ai.h2o.xgboost4j.java.DMatrix dtrain, int iter)

Methods inherited from class java.lang.Object

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

Constructor Detail

BoosterWrapper

public BoosterWrapper(byte[] checkpointBoosterBytes,
                      java.util.Map<java.lang.String,java.lang.Object> params,
                      ai.h2o.xgboost4j.java.DMatrix train)
               throws ai.h2o.xgboost4j.java.XGBoostError

Throws:

ai.h2o.xgboost4j.java.XGBoostError

Method Detail

update

public void update(ai.h2o.xgboost4j.java.DMatrix dtrain,
                   int iter)
            throws ai.h2o.xgboost4j.java.XGBoostError

Throws:

ai.h2o.xgboost4j.java.XGBoostError

saveRabitCheckpoint

public void saveRabitCheckpoint()
                         throws ai.h2o.xgboost4j.java.XGBoostError

Throws:

ai.h2o.xgboost4j.java.XGBoostError

toByteArray

public byte[] toByteArray()
                   throws ai.h2o.xgboost4j.java.XGBoostError

Throws:

ai.h2o.xgboost4j.java.XGBoostError

dispose

public void dispose()

getBooster

public ai.h2o.xgboost4j.java.Booster getBooster()