xgboost
|
Learner interface that integrates objective, gbm and evaluation together. This is the user facing XGBoost training module. More...
#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>
Go to the source code of this file.
Classes | |
class | xgboost::Learner |
Learner class that does training and prediction. This is the user facing module of xgboost training. The Load/Save function corresponds to the model used in python/R. More... | |
struct | xgboost::LearnerModelParam |
Basic model parameters, used to describe the booster. More... | |
Namespaces | |
xgboost | |
namespace of xgboost | |
Enumerations | |
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 } |
Strategy for building multi-target models. More... | |
Learner interface that integrates objective, gbm and evaluation together. This is the user facing XGBoost training module.
Copyright 2015-2023 by XGBoost Contributors