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

Handle formatted JSON input. (#7953)

Browse Source
This commit is contained in:
Jiaming Yuan
2022-06-01 16:20:58 +08:00
committed by GitHub
parent d3429f2ff6
commit 13b15e07e8
3 changed files with 22 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/learner.cc
View File

@@ -852,12 +852,23 @@ class LearnerIO : public LearnerConfiguration {
}
}
// FIXME(jiamingy): Move this out of learner after the old binary model is remove.
auto first_non_space = [&](std::string::const_iterator beg, std::string::const_iterator end) {
for (auto i = beg; i != end; ++i) {
if (!std::isspace(*i)) {
return i;
}
}
return end;
};
if (header[0] == '{') { // Dispatch to JSON
auto buffer = common::ReadAll(fi, &fp);
Json model;
if (header[1] == '"') {
auto it = first_non_space(buffer.cbegin() + 1, buffer.cend());
if (it != buffer.cend() && *it == '"') {
model = Json::Load(StringView{buffer});
} else if (std::isalpha(header[1])) {
} else if (it != buffer.cend() && std::isalpha(*it)) {
model = Json::Load(StringView{buffer}, std::ios::binary);
} else {
LOG(FATAL) << "Invalid model format";