public class XGBoost extends Object
Constructor and Description |
---|
XGBoost() |
Modifier and Type | Method and Description |
---|---|
static String[] |
crossValidation(DMatrix data,
Map<String,Object> params,
int round,
int nfold,
String[] metrics,
IObjective obj,
IEvaluation eval)
Cross-validation with given parameters.
|
static Booster |
loadModel(byte[] buffer)
Load a new Booster model from a byte array buffer.
|
static Booster |
loadModel(InputStream in)
Load a new Booster model from a file opened as input stream.
|
static Booster |
loadModel(String modelPath)
load model from modelPath
|
static Booster |
train(DMatrix dtrain,
Map<String,Object> params,
int round,
Map<String,DMatrix> watches,
float[][] metrics,
IObjective obj,
IEvaluation eval,
int earlyStoppingRound)
Train a booster given parameters.
|
static Booster |
train(DMatrix dtrain,
Map<String,Object> params,
int round,
Map<String,DMatrix> watches,
float[][] metrics,
IObjective obj,
IEvaluation eval,
int earlyStoppingRounds,
Booster booster)
Train a booster given parameters.
|
static Booster |
train(DMatrix dtrain,
Map<String,Object> params,
int round,
Map<String,DMatrix> watches,
IObjective obj,
IEvaluation eval)
Train a booster given parameters.
|
static Booster |
trainAndSaveCheckpoint(DMatrix dtrain,
Map<String,Object> params,
int numRounds,
Map<String,DMatrix> watches,
float[][] metrics,
IObjective obj,
IEvaluation eval,
int earlyStoppingRounds,
Booster booster,
int checkpointInterval,
String checkpointPath,
org.apache.hadoop.fs.FileSystem fs) |
public static Booster loadModel(String modelPath) throws XGBoostError
modelPath
- booster modelPath (model generated by booster.saveModel)XGBoostError
- native errorpublic static Booster loadModel(InputStream in) throws XGBoostError, IOException
in
- The input stream of the file,
will be closed after this function call.XGBoostError
IOException
public static Booster loadModel(byte[] buffer) throws XGBoostError, IOException
buffer
- The byte contents of the booster.XGBoostError
IOException
public static Booster train(DMatrix dtrain, Map<String,Object> params, int round, Map<String,DMatrix> watches, IObjective obj, IEvaluation eval) throws XGBoostError
dtrain
- Data to be trained.params
- Parameters.round
- Number of boosting iterations.watches
- a group of items to be evaluated during training, this allows user to watch
performance on the validation set.obj
- customized objectiveeval
- customized evaluationXGBoostError
public static Booster train(DMatrix dtrain, Map<String,Object> params, int round, Map<String,DMatrix> watches, float[][] metrics, IObjective obj, IEvaluation eval, int earlyStoppingRound) throws XGBoostError
dtrain
- Data to be trained.params
- Parameters.round
- Number of boosting iterations.watches
- a group of items to be evaluated during training, this allows user to watch
performance on the validation set.metrics
- array containing the evaluation metrics for each matrix in watches for each
iterationearlyStoppingRound
- if non-zero, training would be stopped
after a specified number of consecutive
increases in any evaluation metric.obj
- customized objectiveeval
- customized evaluationXGBoostError
public static Booster trainAndSaveCheckpoint(DMatrix dtrain, Map<String,Object> params, int numRounds, Map<String,DMatrix> watches, float[][] metrics, IObjective obj, IEvaluation eval, int earlyStoppingRounds, Booster booster, int checkpointInterval, String checkpointPath, org.apache.hadoop.fs.FileSystem fs) throws XGBoostError, IOException
XGBoostError
IOException
public static Booster train(DMatrix dtrain, Map<String,Object> params, int round, Map<String,DMatrix> watches, float[][] metrics, IObjective obj, IEvaluation eval, int earlyStoppingRounds, Booster booster) throws XGBoostError
dtrain
- Data to be trained.params
- Parameters.round
- Number of boosting iterations.watches
- a group of items to be evaluated during training, this allows user to watch
performance on the validation set.metrics
- array containing the evaluation metrics for each matrix in watches for each
iterationearlyStoppingRounds
- if non-zero, training would be stopped
after a specified number of consecutive
goes to the unexpected direction in any evaluation metric.obj
- customized objectiveeval
- customized evaluationbooster
- train from scratch if set to null; train from an existing booster if not null.XGBoostError
public static String[] crossValidation(DMatrix data, Map<String,Object> params, int round, int nfold, String[] metrics, IObjective obj, IEvaluation eval) throws XGBoostError
data
- Data to be trained.params
- Booster params.round
- Number of boosting iterations.nfold
- Number of folds in CV.metrics
- Evaluation metrics to be watched in CV.obj
- customized objective (set to null if not used)eval
- customized evaluation (set to null if not used)XGBoostError
- native errorCopyright © 2021. All rights reserved.