diff --git a/include/xgboost/c_api.h b/include/xgboost/c_api.h index 68f35ace9..1598aac3b 100644 --- a/include/xgboost/c_api.h +++ b/include/xgboost/c_api.h @@ -266,7 +266,7 @@ XGB_DLL int XGDMatrixNumCol(DMatrixHandle handle, * \param out handle to the result booster * \return 0 when success, -1 when failure happens */ -XGB_DLL int XGBoosterCreate(void* dmats[], +XGB_DLL int XGBoosterCreate(const DMatrixHandle dmats[], bst_ulong len, BoosterHandle *out); /*! diff --git a/src/c_api/c_api.cc b/src/c_api/c_api.cc index d0ea7815b..6ee4f93dc 100644 --- a/src/c_api/c_api.cc +++ b/src/c_api/c_api.cc @@ -463,7 +463,7 @@ int XGDMatrixNumCol(const DMatrixHandle handle, } // xgboost implementation -int XGBoosterCreate(DMatrixHandle dmats[], +int XGBoosterCreate(const DMatrixHandle dmats[], bst_ulong len, BoosterHandle *out) { API_BEGIN();