コード例 #1
0
ファイル: xgboost_python.cpp プロジェクト: AntHar/xgboost
 DLLEXPORT void XGBoosterBoostOneIter( void *handle, void *dtrain, 
                             float *grad, float *hess, size_t len, int bst_group ){
     Booster *bst = static_cast<Booster*>(handle);
     DMatrix *dtr = static_cast<DMatrix*>(dtrain);
     bst->CheckInit(); dtr->CheckInit(); 
     bst->BoostOneIter( *dtr, grad, hess, len, bst_group );
 }      
コード例 #2
0
ファイル: xgboost_wrapper.cpp プロジェクト: nianxue/xgboost
 void XGBoosterBoostOneIter(void *handle, void *dtrain,
                            float *grad, float *hess, bst_ulong len) {
   Booster *bst = static_cast<Booster*>(handle);
   DataMatrix *dtr = static_cast<DataMatrix*>(dtrain);
   bst->CheckInitModel();
   bst->CheckInit(dtr);
   bst->BoostOneIter(*dtr, grad, hess, len);
 }
コード例 #3
0
int XGBoosterBoostOneIter(BoosterHandle handle,
                          DMatrixHandle dtrain,
                          float *grad,
                          float *hess,
                          bst_ulong len) {
  API_BEGIN();
  Booster *bst = static_cast<Booster*>(handle);
  DataMatrix *dtr = static_cast<DataMatrix*>(dtrain);
  bst->CheckInitModel();
  bst->CheckInit(dtr);
  bst->BoostOneIter(*dtr, grad, hess, len);
  API_END();
}