Constructor Detail

• XGBoost

  public XGBoost()

Method Detail

• loadModel

  public static Booster loadModel(String modelPath)
                           throws XGBoostError

  load model from modelPath

  Parameters:

  modelPath - booster modelPath (model generated by booster.saveModel)

  Throws:

  XGBoostError - native error

• loadModel

  public static Booster loadModel(InputStream in)
                           throws XGBoostError,
                                  IOException

  Load a new Booster model from a file opened as input stream. The assumption is the input stream only contains one XGBoost Model. This can be used to load existing booster models saved by other xgboost bindings.

  Parameters:

  in - The input stream of the file, will be closed after this function call.

  Returns:

  The create boosted

  Throws:

  XGBoostError

  IOException

• train

  public static Booster train(DMatrix dtrain,
              Map params,
              int round,
              MapDMatrix> watches,
              IObjective obj,
              IEvaluation eval)
                       throws XGBoostError

  Train a booster with given parameters.

  Parameters:

  dtrain - Data to be trained.

  params - Booster params.

  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 objective (set to null if not used)

  eval - customized evaluation (set to null if not used)

  Returns:

  trained booster

  Throws:

  XGBoostError - native error

• crossValidation

  public static String[] crossValidation(DMatrix data,
                         Map params,
                         int round,
                         int nfold,
                         String[] metrics,
                         IObjective obj,
                         IEvaluation eval)
                                  throws XGBoostError

  Cross-validation with given parameters.

  Parameters:

  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)

  Returns:

  evaluation history

  Throws:

  XGBoostError - native error