包 | 描述 |
---|---|
ml.dmlc.xgboost4j.java |
修饰符和类型 | 方法和描述 |
---|---|
void |
Booster.boost(DMatrix dtrain, float[] grad, float[] hess) 已过时。
|
void |
Booster.boost(DMatrix dtrain, int iter, float[] grad, float[] hess) 使用给定的 grad 和 hess 更新
|
void |
ConfigContext.close() |
static void |
Communicator.communicatorPrint(String msg) 通过通信器打印消息。
|
static String[] |
XGBoost.crossValidation(DMatrix data, Map<String,Object> params, int round, int nfold, String[] metrics, IObjective obj, IEvaluation eval) 使用给定参数进行交叉验证。
|
String |
Booster.evalSet(DMatrix[] evalMatrixs, String[] evalNames, IEvaluation eval) 使用给定的自定义 Evaluation 类进行评估
|
String |
Booster.evalSet(DMatrix[] evalMatrixs, String[] evalNames, IEvaluation eval, float[] metricsOut) |
String |
Booster.evalSet(DMatrix[] evalMatrixs, String[] evalNames, int iter) 使用给定的 dmatrix 进行评估。
|
String |
Booster.evalSet(DMatrix[] evalMatrixs, String[] evalNames, int iter, float[] metricsOut) 使用给定的 dmatrix 进行评估。
|
String |
Booster.getAttr(String key) 从 Booster 获取属性。
|
Map<String,String> |
Booster.getAttrs() 以 Map 形式获取存储在 Booster 中的属性。
|
float[] |
DMatrix.getBaseMargin() 获取 DMatrix 的基准边距 (base margin)。
|
Object |
ConfigContext.getConfig(String name) |
String[] |
Booster.getFeatureNames() 从 Booster 获取特征名称。
|
String[] |
DMatrix.getFeatureNames() 获取特征名称
|
Map<String,Integer> |
Booster.getFeatureScore(String featureMap) 获取每个特征的重要性
|
Map<String,Integer> |
Booster.getFeatureScore(String[] featureNames) 使用指定的特征名称获取每个特征的重要性。
|
String[] |
Booster.getFeatureTypes() 从 Booster 获取特征类型。
|
String[] |
DMatrix.getFeatureTypes() 获取特征类型
|
int[] |
DMatrix.getGroup() 获取 DMatrix 的分组大小
|
float[] |
DMatrix.getLabel() 获取标签值
|
String[] |
Booster.getModelDump(String[] featureNames, boolean withStats) 以字符串数组形式获取模型的转储,并使用指定的特征名称。
|
String[] |
Booster.getModelDump(String[] featureNames, boolean withStats, String format) |
String[] |
Booster.getModelDump(String featureMap, boolean withStats) 以字符串数组形式获取模型的转储
|
String[] |
Booster.getModelDump(String featureMap, boolean withStats, String format) |
int |
Booster.getNumBoostedRound() |
long |
Booster.getNumFeature() 获取模型特征的数量。
|
DMatrix.QuantileCut |
DMatrix.getQuantileCut() 获取分位数切割 (Quantile Cut)。
|
static int |
Communicator.getRank() 获取当前线程的排名。
|
Map<String,Double> |
Booster.getScore(String[] featureNames, String importanceType) 获取增益 (gain) 或覆盖 (cover)(平均或总计)的特征重要性
|
Map<String,Double> |
Booster.getScore(String featureMap, String importanceType) 使用特征名称获取增益 (gain) 或覆盖 (cover)(平均或总计)的特征重要性
|
float[] |
DMatrix.getWeight() 获取 DMatrix 的权重
|
Map<String,Object> |
RabitTracker.getWorkerArgs() 获取可用于传递给工作者的环境。
|
Map<String,Object> |
ITracker.getWorkerArgs() |
static int |
Communicator.getWorldSize() 获取当前作业的世界大小 (world size)。
|
static void |
Communicator.init(Map<String,Object> envs) 在当前工作线程上初始化集体通信器 (collective communicator)。
|
float[][] |
Booster.inplace_predict(float[] data, int nrow, int ncol, float missing) 执行线程安全的预测。
|
float[][] |
Booster.inplace_predict(float[] data, int nrow, int ncol, float missing, int[] iteration_range) 执行线程安全的预测。
|
float[][] |
Booster.inplace_predict(float[] data, int nrow, int ncol, float missing, int[] iteration_range, Booster.PredictionType predict_type, float[] base_margin) 执行线程安全的预测。
|
Booster |
ExternalCheckpointManager.loadCheckpointAsBooster() |
static Booster |
XGBoost.loadModel(byte[] buffer) 从字节数组缓冲区加载新的 Booster 模型。
|
static Booster |
XGBoost.loadModel(InputStream in) 从以输入流形式打开的文件加载新的 Booster 模型。
|
static Booster |
XGBoost.loadModel(String modelPath) 从 modelPath 加载模型
|
long |
DMatrix.nonMissingNum() 获取 DMatrix 中非缺失值的数量。
|
float[][] |
Booster.predict(DMatrix data) 使用数据进行预测
|
float[][] |
Booster.predict(DMatrix data, boolean outputMargin) 使用数据进行预测
|
float[][] |
Booster.predict(DMatrix data, boolean outputMargin, int treeLimit) 具有所有选项的高级预测函数。
|
float[][] |
Booster.predictContrib(DMatrix data, int treeLimit) 输出特征对给定数据预测的贡献
|
float[][] |
Booster.predictLeaf(DMatrix data, int treeLimit) 给定数据,预测叶子索引
|
long |
DMatrix.rowNum() 获取 DMatrix 的行数
|
void |
Booster.saveModel(OutputStream out) 将模型保存到以输出流形式打开的文件。
|
void |
Booster.saveModel(OutputStream out, String format) 将模型保存到以输出流形式打开的文件。
|
void |
Booster.saveModel(String modelPath) 将模型保存到 modelPath
|
void |
Booster.setAttr(String key, String value) 为 Booster 设置属性。
|
void |
Booster.setAttrs(Map<String,String> attrs) 为 Booster 设置属性。
|
void |
DMatrix.setBaseMargin(Column column) 从数组接口设置 DMatrix 的基准边距 (base margin)
|
void |
DMatrix.setBaseMargin(float[] baseMargin) 设置基准边距 (base margin)(初始预测)。
|
void |
DMatrix.setBaseMargin(float[][] baseMargin) 设置基准边距 (base margin)(初始预测)。
|
void |
ConfigContext.setConfig(String key, Object value) 设置单个配置
|
void |
ConfigContext.setConfigs(Map<String,Object> configs) 设置一组配置
|
void |
Booster.setFeatureNames(String[] featureNames) 为 Booster 设置特征名称。
|
void |
DMatrix.setFeatureNames(String[] values) 设置特征名称
|
void |
Booster.setFeatureTypes(String[] featureTypes) 为 Booster 设置特征类型。
|
void |
DMatrix.setFeatureTypes(String[] values) 设置特征类型
|
void |
DMatrix.setGroup(int[] group) 设置 DMatrix 的分组大小(用于排序)
|
void |
DMatrix.setLabel(Column column) 从数组接口设置 DMatrix 的标签
|
void |
DMatrix.setLabel(float[] labels) 设置 dmatrix 的标签
|
void |
Booster.setParam(String key, Object value) 为 Booster 设置参数。
|
void |
Booster.setParams(Map<String,Object> params) 为 Booster 设置参数。
|
void |
DMatrix.setQueryId(Column column) 从数组接口设置 DMatrix 的查询 ID
|
void |
DMatrix.setQueryId(int[] qid) 设置查询 ID(用于排序)
|
void |
DMatrix.setWeight(Column column) 从数组接口设置 DMatrix 的权重
|
void |
DMatrix.setWeight(float[] weights) 设置每个实例的权重
|
static void |
Communicator.shutdown() 关闭当前工作线程中的通信器,相当于 finalize。
|
DMatrix |
DMatrix.slice(int[] rowIndex) 切片 DMatrix 并返回一个仅包含 `rowIndex` 的新 DMatrix。
|
boolean |
RabitTracker.start() |
boolean |
ITracker.start() |
void |
RabitTracker.stop() |
void |
ITracker.stop() |
byte[] |
Booster.toByteArray() 将模型保存为 UBJSON ("ubj") 格式的原始字节数组。
|
byte[] |
Booster.toByteArray(String format) 将模型保存为原始字节数组。
|
static Booster |
XGBoost.train(DMatrix dtrain, Map<String,Object> params, int round, Map<String,DMatrix> watches, float[][] metrics, IObjective obj, IEvaluation eval, int earlyStoppingRound) 使用给定参数训练 Booster。
|
static Booster |
XGBoost.train(DMatrix dtrain, Map<String,Object> params, int round, Map<String,DMatrix> watches, float[][] metrics, IObjective obj, IEvaluation eval, int earlyStoppingRounds, Booster booster) 使用给定参数训练 Booster。
|
static Booster |
XGBoost.train(DMatrix dtrain, Map<String,Object> params, int round, Map<String,DMatrix> watches, IObjective obj, IEvaluation eval) 使用给定参数训练 Booster。
|
static Booster |
XGBoost.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) |
void |
Booster.update(DMatrix dtrain, int iter) 更新 booster 一次迭代。
|
void |
Booster.update(DMatrix dtrain, int iter, IObjective obj) 使用自定义目标函数更新
|
void |
Booster.update(DMatrix dtrain, IObjective obj) 已过时。
|
void |
ExternalCheckpointManager.updateCheckpoint(Booster boosterToCheckpoint) |
void |
RabitTracker.waitFor(long timeout) |
void |
ITracker.waitFor(long taskExecutionTimeout) |
构造函数和描述 |
---|
ConfigContext() |
ConfigContext(Map<String,Object> params) |
DMatrix(BigDenseMatrix matrix)
从 BigDenseMatrix 创建 DMatrix
|
DMatrix(BigDenseMatrix matrix, float missing)
从密集矩阵创建 DMatrix
|
DMatrix(ColumnBatch columnBatch, float missing, int nthread)
从列数组接口创建普通的 DMatrix
|
DMatrix(float[] data, int nrow, int ncol)
已过时。
请使用
DMatrix(float[], int, int, float) 明确指定缺失值 |
DMatrix(float[] data, int nrow, int ncol, float missing)
从密集矩阵创建 DMatrix
|
DMatrix(Iterator<LabeledPoint> iter, String cacheInfo)
从迭代器创建 DMatrix。
|
DMatrix(Iterator<LabeledPoint> iter, String cacheInfo, float missing)
从迭代器创建 DMatrix。
|
DMatrix(long[] headers, int[] indices, float[] data, DMatrix.SparseType st)
已过时。
|
DMatrix(long[] headers, int[] indices, float[] data, DMatrix.SparseType st, int shapeParam)
从 CSR/CSC 格式的稀疏矩阵创建 DMatrix。
|
DMatrix(long[] headers, int[] indices, float[] data, DMatrix.SparseType st, int shapeParam, float missing, int nthread) |
DMatrix(String dataPath)
通过从 dataPath 加载 libsvm 文件创建 DMatrix
|
ExternalCheckpointManager(String checkpointPath, org.apache.hadoop.fs.FileSystem fs) |
RabitTracker(int numWorkers) |
RabitTracker(int numWorkers, String hostIp) |
RabitTracker(int numWorkers, String hostIp, int port, int timeout) |
Copyright © 2025. 保留所有权利。