ml.dmlc.xgboost4j.java

XGBoost 类

java.lang.Object

ml.dmlc.xgboost4j.java.XGBoost

public class XGBoost
extends Object

xgboost 训练器

作者

hzx

字段摘要

字段

修饰符和类型	字段和说明
静态 String[]	MAXIMIZE_METRICES

构造器摘要

构造器

构造器和说明
XGBoost()

方法摘要

修饰符和类型	方法和说明
静态 String[]	crossValidation(DMatrix data, Map<String,Object> params, int round, int nfold, String[] metrics, IObjective obj, IEvaluation eval) 使用给定参数进行交叉验证。
静态布尔值	isMaximizeEvaluation(String evalInfo, String[] evalNames, Map<String,Object> params)
静态 Booster	loadModel(byte[] buffer) 从字节数组缓冲区加载新的 Booster 模型。
静态 Booster	loadModel(InputStream in) 从作为输入流打开的文件加载新的 Booster 模型。
静态 Booster	loadModel(String modelPath) 从 modelPath 加载模型
静态 Booster	train(DMatrix dtrain, Map<String,Object> params, int round, Map<String,DMatrix> watches, float[][] metrics, IObjective obj, IEvaluation eval, int earlyStoppingRound) 训练一个给定参数的 Booster。
静态 Booster	train(DMatrix dtrain, Map<String,Object> params, int round, Map<String,DMatrix> watches, float[][] metrics, IObjective obj, IEvaluation eval, int earlyStoppingRounds, Booster booster) 训练一个给定参数的 Booster。
静态 Booster	train(DMatrix dtrain, Map<String,Object> params, int round, Map<String,DMatrix> watches, IObjective obj, IEvaluation eval) 训练一个给定参数的 Booster。
静态 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)

从类 java.lang.Object继承的方法

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

字段详细信息

MAXIMIZE_METRICES

public static final String[] MAXIMIZ_METRICES

构造器详细信息

XGBoost

public XGBoost()

方法详细信息

加载模型

public static Booster loadModel(String modelPath)
                         throws XGBoostError

从 modelPath 加载模型

参数

modelPath - Booster 模型路径（由 booster.saveModel 生成的模型）

抛出

XGBoostError - 原生错误

加载模型

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

从作为输入流打开的文件加载新的 Booster 模型。假设输入流只包含一个 XGBoost 模型。这可以用于加载由其他 XGBoost 绑定保存的现有 Booster 模型。

参数

in - 文件的输入流，此函数调用后将关闭。

返回

创建已提升的模型

抛出

XGBoostError

IOException

加载模型

public static Booster loadModel(byte[] buffer)
                         throws XGBoostError,
                                IOException

从字节数组缓冲区加载新的 Booster 模型。假设数组只包含一个 XGBoost 模型。这可以用于加载由其他 XGBoost 绑定保存的现有 Booster 模型。

参数

buffer - 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

训练一个给定参数的 Booster。

参数

dtrain - 要训练的数据。

params - 参数。

round - 提升迭代次数。

watches - 在训练期间要评估的一组项，这允许用户观察验证集上的性能。

obj - 自定义目标

eval - 自定义评估

返回

训练好的 Booster。

抛出

XGBoostError

训练

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

训练一个给定参数的 Booster。

参数

dtrain - 要训练的数据。

params - 参数。

round - 提升迭代次数。

watches - 在训练期间要评估的一组项，这允许用户观察验证集上的性能。

metrics - 包含每次迭代中 watches 中每个矩阵的评估指标的数组

earlyStoppingRound - 如果非零，则在任何评估指标连续增加指定次数后，训练将停止。

obj - 自定义目标

eval - 自定义评估

返回

训练好的 Booster。

抛出

XGBoostError

训练并保存检查点

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

训练一个给定参数的 Booster。

参数

dtrain - 要训练的数据。

params - 参数。

round - 提升迭代次数。

watches - 在训练期间要评估的一组项，这允许用户观察验证集上的性能。

metrics - 包含每次迭代中 watches 中每个矩阵的评估指标的数组

earlyStoppingRounds - 如果非零，则在任何评估指标连续朝意外方向发展指定次数后，训练将停止。

obj - 自定义目标

eval - 自定义评估

booster - 如果设置为 null，则从头开始训练；如果非 null，则从现有 Booster 开始训练。

返回

训练好的 Booster。

抛出

XGBoostError

isMaximizeEvaluation

public static boolean isMaximizeEvaluation(String evalInfo,
                                           String[] evalNames,
                                           Map<String,Object> params)

交叉验证

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

使用给定参数进行交叉验证。

参数

data - 要训练的数据。

params - Booster 参数。

round - 提升迭代次数。

nfold - 交叉验证中的折叠数。

metrics - 在交叉验证中要观察的评估指标。

obj - 自定义目标（如果未使用，则设置为 null）

eval - 自定义评估（如果未使用，则设置为 null）

返回

评估历史

抛出

XGBoostError - 原生错误