Clarify meaning of training parameter in XGBoosterPredict() (#5604)

Co-authored-by: Hyunsu Cho <chohyu01@cs.washington.edu> Co-authored-by: Jiaming Yuan <jm.yuan@outlook.com>
2020-04-25 19:48:42 -04:00
parent 4fd95272c8
commit 8dfe7b3686
2 changed files with 9 additions and 2 deletions
--- a/src/c_api/c_api.cc
+++ b/src/c_api/c_api.cc
@@ -426,7 +426,7 @@ XGB_DLL int XGBoosterPredict(BoosterHandle handle,
                             DMatrixHandle dmat,
                             int option_mask,
                             unsigned ntree_limit,
-                             int32_t training,
+                             int training,
                             xgboost::bst_ulong *len,
                             const bst_float **out_result) {
  API_BEGIN();