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

Fix typo. (#7680)

Browse Source
This commit is contained in:
Jiaming Yuan
2022-02-20 03:42:47 +08:00
committed by GitHub
parent b2341eab0c
commit 671b3c8d8e
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/tree/tree_model.cc
View File

@@ -868,7 +868,8 @@ void RegTree::Save(dmlc::Stream* fo) const {
CHECK_EQ(param.num_nodes, static_cast<int>(stats_.size()));
CHECK_EQ(param.deprecated_num_roots, 1);
CHECK_NE(param.num_nodes, 0);
CHECK(!HasCategoricalSplit()) << "Please JSON/UBJSON for saving models with categorical splits.";
CHECK(!HasCategoricalSplit())
<< "Please use JSON/UBJSON for saving models with categorical splits.";
if (DMLC_IO_NO_ENDIAN_SWAP) {
fo->Write(&param, sizeof(TreeParam));