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

half ram support

Browse Source
This commit is contained in:
tqchen
2015-04-19 21:29:13 -07:00
parent 5ad1555daf
commit dfec406afd
3 changed files with 32 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/learner/learner-inl.hpp
View File

@@ -260,9 +260,9 @@ class BoostLearner : public rabit::Serializable {
std::vector<bool> enabled(ncol, true);
// initialize column access
p_train->fmat()->InitColAccess(enabled, prob_buffer_row);
const int kMagicSimple = 0xffffab01;
// check, if it is not DMatrix simple, then use hist maker
if (p_train->magic != kMagicSimple) {
const int kMagicPage = 0xffffab02;
// check, if it is DMatrixPage, then use hist maker
if (p_train->magic == kMagicPage) {
this->SetParam("updater", "grow_histmaker,prune");
}
}