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

Check external memory support for exact tree method. (#7088)

Browse Source
This commit is contained in:
Jiaming Yuan 2021-07-08 02:12:57 +08:00 committed by GitHub
parent 615ab2b03e
commit 689eb8f620
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/tree/updater_colmaker.cc
View File

@ -98,6 +98,10 @@ class ColMaker: public TreeUpdater {
LOG(FATAL) << "Updater `grow_colmaker` or `exact` tree method doesn't " LOG(FATAL) << "Updater `grow_colmaker` or `exact` tree method doesn't "
"support distributed training."; "support distributed training.";
} }
if (!dmat->SingleColBlock()) {
LOG(FATAL) << "Updater `grow_colmaker` or `exact` tree method doesn't "
"support external memory training.";
}
this->LazyGetColumnDensity(dmat); this->LazyGetColumnDensity(dmat);
// rescale learning rate according to size of trees // rescale learning rate according to size of trees
float lr = param_.learning_rate; float lr = param_.learning_rate;