xgboost
|
整合了目标函数、gbm 和评估的 Learner 接口。这是面向用户的 XGBoost 训练模块。 更多...
#include <dmlc/io.h>
#include <xgboost/base.h>
#include <xgboost/context.h>
#include <xgboost/linalg.h>
#include <xgboost/metric.h>
#include <xgboost/model.h>
#include <xgboost/span.h>
#include <xgboost/task.h>
#include <algorithm>
#include <cstdint>
#include <map>
#include <memory>
#include <string>
#include <utility>
#include <vector>
类 | |
class | xgboost::Learner |
Learner 类,用于执行训练和预测。这是面向用户的 xgboost 训练模块。Load/Save 函数对应于 python/R 中使用的模型。 更多... | |
struct | xgboost::LearnerModelParam |
基本模型参数,用于描述 booster。 更多... | |
命名空间 | |
xgboost | |
多目标树的核心数据结构。 | |
枚举 | |
enum class | xgboost::PredictionType : std::uint8_t { xgboost::kValue = 0 , xgboost::kMargin = 1 , xgboost::kContribution = 2 , xgboost::kApproxContribution = 3 , xgboost::kInteraction = 4 , xgboost::kApproxInteraction = 5 , xgboost::kLeaf = 6 } |
enum class | xgboost::MultiStrategy : std::int32_t { xgboost::kOneOutputPerTree = 0 , xgboost::kMultiOutputTree = 1 } |
构建多目标模型的策略。 更多... | |
整合了目标函数、gbm 和评估的 Learner 接口。这是面向用户的 XGBoost 训练模块。
Copyright 2015-2023 由 XGBoost 贡献者