From e055dc153e06538142420623aa51cc9939a54cac Mon Sep 17 00:00:00 2001
From: fis <jm.yuan@outlook.com>
Date: Wed, 30 Sep 2020 19:55:40 +0800
Subject: [PATCH] Add JSON schema for categorical splits.

---
 doc/model.schema | 30 +++++++++++++++++++++++++++++-
 1 file changed, 29 insertions(+), 1 deletion(-)

diff --git a/doc/model.schema b/doc/model.schema
index 9752745975c7..6c7bd9c32fd6 100644
--- a/doc/model.schema
+++ b/doc/model.schema
@@ -93,6 +93,30 @@
                     "items": {
                       "type": "boolean"
                     }
+                  },
+                  "categories": {
+                    "type": "array",
+                    "items": {
+                      "type": "integer"
+                    }
+                  },
+                  "categories_nodes": {
+                    "type": "array",
+                    "items": {
+                      "type": "integer"
+                    }
+                  },
+                  "categories_segments": {
+                    "type": "array",
+                    "items": {
+                      "type": "integer"
+                    }
+                  },
+                  "categorical_sizes": {
+                    "type": "array",
+                    "items": {
+                      "type": "integer"
+                    }
                   }
                 },
                 "required": [
@@ -105,7 +129,11 @@
                   "parents",
                   "split_indices",
                   "split_conditions",
-                  "default_left"
+                  "default_left",
+                  "categories",
+                  "categories_nodes",
+                  "categories_segments",
+                  "categories_sizes"
                 ]
               }
             },