xgboost
Classes | Namespaces | Enumerations
learner.h File Reference

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/generic_parameters.h>
#include <xgboost/host_device_vector.h>
#include <xgboost/model.h>
#include <xgboost/predictor.h>
#include <xgboost/task.h>
#include <map>
#include <memory>
#include <string>
#include <utility>
#include <vector>
Include dependency graph for learner.h:

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
}
 

Detailed Description

Learner interface that integrates objective, gbm and evaluation together. This is the user facing XGBoost training module.

Copyright 2015-2022 by XGBoost Contributors

Author
Tianqi Chen