From ddc4f20e547b5a23562666a11cddb943f7e97a70 Mon Sep 17 00:00:00 2001
From: Jiaming Yuan <jm.yuan@outlook.com>
Date: Wed, 7 Oct 2020 17:33:31 +0800
Subject: [PATCH] Add JSON schema for categorical splits. (#6194)

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

diff --git a/doc/model.schema b/doc/model.schema
index 975274597..6c7bd9c32 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"
                 ]
               }
             },