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/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>
Include dependency graph for learner.h:

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...
 

Detailed Description

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

Copyright 2015-2023 by XGBoost Contributors

Author
Tianqi Chen