Add JSON schema for categorical splits. (#6194)

2020-10-07 17:33:31 +08:00
parent a2fea33103
commit ddc4f20e54
1 changed files with 29 additions and 1 deletions
--- 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"
                ]
              }
            },