Skip to content

Commit

Permalink
Cleanup.
Browse files Browse the repository at this point in the history
  • Loading branch information
trivialfis committed Jun 15, 2021
1 parent d6ac445 commit 0c2397e
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 3 deletions.
2 changes: 0 additions & 2 deletions include/xgboost/learner.h
Original file line number Diff line number Diff line change
Expand Up @@ -46,8 +46,6 @@ struct XGBAPIThreadLocalEntry {
std::string ret_str;
/*! \brief result holder for returning strings */
std::vector<std::string> ret_vec_str;
/*! \brief result holder for returning unsigned integers */
std::vector<uint32_t> ret_vec_uint32;
/*! \brief result holder for returning string pointers */
std::vector<const char *> ret_vec_charp;
/*! \brief returning float vector. */
Expand Down
2 changes: 1 addition & 1 deletion src/c_api/c_api.cc
Original file line number Diff line number Diff line change
Expand Up @@ -1115,7 +1115,7 @@ XGB_DLL int XGBoosterFeatureScore(BoosterHandle handle,
FeatureMap feature_map = LoadFeatureMap(feature_map_uri);

auto& scores = learner->GetThreadLocal().ret_vec_float;
auto& features = learner->GetThreadLocal().ret_vec_uint32;
std::vector<bst_feature_t> features;
learner->CalcFeatureScore(importance, &features, &scores);

auto n_features = learner->GetNumFeature();
Expand Down

0 comments on commit 0c2397e

Please sign in to comment.