xgboost
|
Learner interface that integrates objective, gbm and evaluation together. This is the user facing XGBoost training module. More...
#include <dmlc/any.h>
#include <xgboost/base.h>
#include <xgboost/feature_map.h>
#include <xgboost/predictor.h>
#include <xgboost/generic_parameters.h>
#include <xgboost/host_device_vector.h>
#include <xgboost/model.h>
#include <utility>
#include <map>
#include <memory>
#include <string>
#include <vector>
Go to the source code of this file.
Classes | |
struct | xgboost::XGBAPIThreadLocalEntry |
entry to to easily hold returning information More... | |
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 |
Namespaces | |
xgboost | |
namespace of xgboost | |
Enumerations | |
enum | xgboost::PredictionType : std::uint8_t { xgboost::PredictionType::kValue = 0, xgboost::PredictionType::kMargin = 1, xgboost::PredictionType::kContribution = 2, xgboost::PredictionType::kApproxContribution = 3, xgboost::PredictionType::kInteraction = 4, xgboost::PredictionType::kApproxInteraction = 5, xgboost::PredictionType::kLeaf = 6 } |
Learner interface that integrates objective, gbm and evaluation together. This is the user facing XGBoost training module.
Copyright 2015-2021 by Contributors