| 包 | 描述 |
|---|---|
| ml.dmlc.xgboost4j.java |
| 修饰符和类型 | 方法和说明 |
|---|---|
void |
Booster.boost(DMatrix dtrain, float[] grad, float[] hess)已弃用。
|
void |
Booster.boost(DMatrix dtrain, int iter, float[] grad, float[] hess)使用给定的梯度和Hessian更新
|
void |
ConfigContext.close() |
static void |
Communicator.communicatorPrint(String msg)通过通信器打印消息。
|
静态 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)使用自定义评估类评估Booster模型
|
String |
Booster.evalSet(DMatrix[] evalMatrixs, String[] evalNames, IEvaluation eval, float[] metricsOut) |
String |
Booster.evalSet(DMatrix[] evalMatrixs, String[] evalNames, int iter)使用给定的DMatrix评估Booster模型。
|
String |
Booster.evalSet(DMatrix[] evalMatrixs, String[] evalNames, int iter, float[] metricsOut)使用给定的DMatrix评估Booster模型。
|
String |
Booster.getAttr(String key)根据提供的键从Booster中获取属性值。
|
Map<String,String> |
Booster.getAttrs()获取存储在Booster中的属性,以Map形式返回。
|
float[] |
DMatrix.getBaseMargin()获取 DMatrix 的基本边距。
|
Object |
ConfigContext.getConfig(String name) |
String[] |
DMatrix.getFeatureNames()获取特征名称
|
String[] |
Booster.getFeatureNames()从Booster中获取所有特征名称。
|
Map<String,Integer> |
Booster.getFeatureScore(String featureMap)获取每个特征的重要性
|
Map<String,Integer> |
Booster.getFeatureScore(String[] featureNames)获取具有指定特征名称的每个特征的重要性。
|
String[] |
DMatrix.getFeatureTypes()获取特征类型
|
String[] |
Booster.getFeatureTypes()从Booster获取特征类型。
|
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()获取分位数切点。
|
static int |
Communicator.getRank()获取当前线程的排名。
|
Map<String,Double> |
Booster.getScore(String[] featureNames, String importanceType)获取增益或覆盖(平均或总和)的特征重要性
|
Map<String,Double> |
Booster.getScore(String featureMap, String importanceType)获取增益或覆盖(平均或总和)的特征重要性,带有特征名称
|
float[] |
DMatrix.getWeight()获取 DMatrix 的权重
|
Map<String,Object> |
RabitTracker.getWorkerArgs()获取可用于传递给工作器的环境。
|
Map<String,Object> |
ITracker.getWorkerArgs() |
static int |
Communicator.getWorldSize()获取当前作业的世界大小。
|
static void |
Communicator.init(Map<String,Object> envs)初始化当前工作线程上的集体通信器。
|
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)使用DMatrix格式的测试数据进行预测。
|
float[][] |
Booster.predict(DMatrix data, boolean outputMargin)使用DMatrix格式的测试数据和输出边距进行预测。
|
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 的基本边距
|
void |
DMatrix.setBaseMargin(float[] baseMargin)设置基本边距(初始预测)。
|
void |
DMatrix.setBaseMargin(float[][] baseMargin)设置基本边距(初始预测)。
|
void |
ConfigContext.setConfig(String key, Object value)设置单个配置
|
void |
ConfigContext.setConfigs(Map<String,Object> configs)设置一堆配置
|
void |
DMatrix.setFeatureNames(String[] values)设置特征名称
|
void |
Booster.setFeatureNames(String[] featureNames)为Booster设置特征名称。
|
void |
DMatrix.setFeatureTypes(String[] values)设置特征类型
|
void |
Booster.setFeatureTypes(String[] featureTypes)为Booster设置特征类型。
|
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。
|
布尔值 |
RabitTracker.start() |
布尔值 |
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)更新助推器一次迭代。
|
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) |
版权所有 © 2025. 保留所有权利。