learner.h File Reference

#include <dmlc/io.h>
#include <xgboost/base.h>
#include <xgboost/context.h>
#include <xgboost/gradient.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
	Learner interface that integrates objective, gbm and evaluation together. This is the user facing XGBoost training module.

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