learner.h File Reference

Copyright 2015-2023 by XGBoost Contributors. 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.

Data Structures
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
namespace	xgboost
	namespace of xgboost

Enumerations
enum class	PredictionType : std::uint8_t {   kValue = 0 , kMargin = 1 , kContribution = 2 , kApproxContribution = 3 ,   kInteraction = 4 , kApproxInteraction = 5 , kLeaf = 6 }
enum class	xgboost::MultiStrategy : std::int32_t { kOneOutputPerTree = 0 , kMultiOutputTree = 1 }
	Strategy for building multi-target models.

Detailed Description

Copyright 2015-2023 by XGBoost Contributors.

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

Author

Tianqi Chen