MR_LIBS/cpu__treeshap_8h_source.html

#ifndef XGBOOST_PREDICTOR_CPU_TREESHAP_H_

#define XGBOOST_PREDICTOR_CPU_TREESHAP_H_

#include <vector>                // vector


#include "xgboost/tree_model.h"  // RegTree


namespace xgboost {

void CalculateContributions(RegTree const &tree, const RegTree::FVec &feat,

                            std::vector<float> *mean_values, bst_float *out_contribs, int condition,

                            unsigned condition_feature);

}  // namespace xgboost

#endif  // XGBOOST_PREDICTOR_CPU_TREESHAP_H_

xgboost::RegTree
define regression tree to be the most common tree model.
Definition tree_model.h:158

xgboost
namespace of xgboost
Definition base.h:90

xgboost::bst_float
float bst_float
float type, used for storing statistics
Definition base.h:97

xgboost::RegTree::FVec
dense feature vector that can be taken by RegTree and can be construct from sparse feature vector.
Definition tree_model.h:554

tree_model.h
Copyright 2014-2023 by Contributors.