▼ include | |
▼ xgboost | |
base.h | Defines configuration macros of xgboost |
c_api.h | C API of XGBoost, used for interfacing to other languages |
data.h | The input data structure of xgboost |
feature_map.h | Feature map data structure to help visualization and model dump |
gbm.h | Interface of gradient booster, that learns through gradient statistics |
generic_parameters.h | |
global_config.h | Global configuration for XGBoost |
host_device_vector.h | A device-and-host vector abstraction layer |
intrusive_ptr.h | Implementation of Intrusive Ptr |
json.h | |
json_io.h | |
learner.h | Learner interface that integrates objective, gbm and evaluation together. This is the user facing XGBoost training module |
linalg.h | Linear algebra related utilities |
linear_updater.h | |
metric.h | Interface of evaluation metric function supported in xgboost |
model.h | Defines the abstract interface for different components in XGBoost |
objective.h | Interface of objective function used by xgboost |
parameter.h | Macro for using C++11 enum class as DMLC parameter |
predictor.h | Interface of predictor, performs predictions for a gradient booster |
span.h | |
string_view.h | |
task.h | |
tree_model.h | Model structure for tree |
tree_updater.h | General primitive for tree learning, Updating a collection of trees given the information |
version_config.h | |
▼ src | |
▼ common | |
base64.h | Data stream support to input and output from/to base64 stream base64 is easier to store and pass as text format in mapreduce |
bitfield.h | |
categorical.h | |
charconv.h | |
column_matrix.h | Utility for fast column-wise access |
common.h | Common utilities |
compressed_iterator.h | |
config.h | Helper class to load in configures from file |
group_data.h | This file defines utils to group data by integer keys Input: given input sequence (key,value), (k1,v1), (k2,v2) Ouptupt: an array of values data = [v1,v2,v3 .. vn] and a group pointer ptr, data[ptr[k]:ptr[k+1]] contains values that corresponds to key k |
hist_util.h | Utility for fast histogram aggregation |
io.h | General stream interface for serialization, I/O |
linalg_op.h | |
math.h | Additional math utils |
observer.h | |
partition_builder.h | |
probability_distribution.h | Implementation of a few useful probability distributions |
pseudo_huber.h | |
quantile.h | Util to compute quantiles |
random.h | Utility related to random |
row_set.h | Quick Utility to compute subset of rows |
survival_util.h | Utility functions, useful for implementing objective and metric functions for survival analysis |
threading_utils.h | |
timer.h | |
transform.h | |
version.h | |