Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

chg root index to booster info, need review

Browse Source
This commit is contained in:
tqchen
2014-08-22 16:26:37 -07:00
parent 58d74861b9
commit 58354643b0
10 changed files with 48 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
python/xgboost_wrapper.cpp
View File

@@ -37,7 +37,7 @@ class Booster: public learner::BoostLearner<FMatrixS> {
for (unsigned j = 0; j < ndata; ++j) {
gpair_[j] = bst_gpair(grad[j], hess[j]);
}
gbm_->DoBoost(gpair_, train.fmat, train.info.root_index);
gbm_->DoBoost(gpair_, train.fmat, train.info.info);
}
inline void CheckInitModel(void) {
if (!init_model) {
@@ -151,8 +151,8 @@ extern "C"{
if (src.info.weights.size() != 0) {
ret.info.weights.push_back(src.info.weights[ridx]);
}
if (src.info.root_index.size() != 0) {
ret.info.weights.push_back(src.info.root_index[ridx]);
if (src.info.info.root_index.size() != 0) {
ret.info.info.root_index.push_back(src.info.info.root_index[ridx]);
}
}
return p_ret;