Ensure models with categorical splits don't use old binary format. (#7666)

2022-02-19 08:05:28 +08:00
parent 14d61b0141
commit 7366d3b20c
3 changed files with 25 additions and 1 deletions
--- a/src/tree/tree_model.cc
+++ b/src/tree/tree_model.cc
@@ -868,6 +868,7 @@ 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.";

  if (DMLC_IO_NO_ENDIAN_SWAP) {
    fo->Write(&param, sizeof(TreeParam));