Improve JSON format for categorical features. (#6128)

* Gather categories for all nodes.
This commit is contained in:
Jiaming Yuan
2020-09-21 15:35:05 +08:00
committed by GitHub
parent 210c131ce7
commit 7065779afa
2 changed files with 113 additions and 51 deletions

View File

@@ -620,6 +620,8 @@ class RegTree : public Model {
};
private:
void LoadCategoricalSplit(Json const& in);
void SaveCategoricalSplit(Json* p_out) const;
// vector of nodes
std::vector<Node> nodes_;
// free node space, used during training process