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

fix DMatrix load_row_split bug (#3431)

Browse Source
This commit is contained in:
hlsc 2018-07-29 08:21:30 +08:00 committed by Philip Hyunsu Cho
parent 8973f2cb0e
commit 5850a2558a
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/c_api/c_api.cc
View File

@ -213,11 +213,13 @@ int XGDMatrixCreateFromFile(const char *fname,
int silent, int silent,
DMatrixHandle *out) { DMatrixHandle *out) {
API_BEGIN(); API_BEGIN();
bool load_row_split = false;
if (rabit::IsDistributed()) { if (rabit::IsDistributed()) {
LOG(CONSOLE) << "XGBoost distributed mode detected, " LOG(CONSOLE) << "XGBoost distributed mode detected, "
<< "will split data among workers"; << "will split data among workers";
load_row_split = true;
} }
*out = new std::shared_ptr<DMatrix>(DMatrix::Load(fname, silent != 0, true)); *out = new std::shared_ptr<DMatrix>(DMatrix::Load(fname, silent != 0, load_row_split));
API_END(); API_END();
} }