public class BoosterWrapper
extends java.lang.Object
| Constructor and Description |
|---|
BoosterWrapper(byte[] checkpointBoosterBytes,
java.util.Map<java.lang.String,java.lang.Object> params,
ml.dmlc.xgboost4j.java.DMatrix train) |
| Modifier and Type | Method and Description |
|---|---|
void |
dispose() |
ml.dmlc.xgboost4j.java.Booster |
getBooster() |
void |
saveRabitCheckpoint() |
byte[] |
toByteArray() |
void |
update(ml.dmlc.xgboost4j.java.DMatrix dtrain,
int iter) |
public BoosterWrapper(byte[] checkpointBoosterBytes,
java.util.Map<java.lang.String,java.lang.Object> params,
ml.dmlc.xgboost4j.java.DMatrix train)
throws ml.dmlc.xgboost4j.java.XGBoostError
ml.dmlc.xgboost4j.java.XGBoostErrorpublic void update(ml.dmlc.xgboost4j.java.DMatrix dtrain,
int iter)
throws ml.dmlc.xgboost4j.java.XGBoostError
ml.dmlc.xgboost4j.java.XGBoostErrorpublic void saveRabitCheckpoint()
throws ml.dmlc.xgboost4j.java.XGBoostError
ml.dmlc.xgboost4j.java.XGBoostErrorpublic byte[] toByteArray()
throws ml.dmlc.xgboost4j.java.XGBoostError
ml.dmlc.xgboost4j.java.XGBoostErrorpublic void dispose()
public ml.dmlc.xgboost4j.java.Booster getBooster()