Add XGBoosterGetNumFeature (#5856)

- add GetNumFeature to Learner
- add XGBoosterGetNumFeature to C API
- update c-api-demo accordingly
This commit is contained in:
Alexander Gugel
2020-07-14 07:25:17 +01:00
committed by GitHub
parent e0c179c7cc
commit 970b4b3fa2
6 changed files with 36 additions and 1 deletions

View File

@@ -60,6 +60,10 @@ int main(int argc, char** argv) {
printf("%s\n", eval_result);
}
bst_ulong num_feature = 0;
safe_xgboost(XGBoosterGetNumFeature(booster, &num_feature));
printf("num_feature: %llu\n", num_feature);
// predict
bst_ulong out_len = 0;
const float* out_result = NULL;