Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add JSON schema for categorical splits. (#6194)

Browse Source
This commit is contained in:
Jiaming Yuan 2020-10-07 17:33:31 +08:00 committed by GitHub
parent a2fea33103
commit ddc4f20e54
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 29 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
doc/model.schema
View File

@ -93,6 +93,30 @@
"items": { "items": {
"type": "boolean" "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": [ "required": [
@ -105,7 +129,11 @@
"parents", "parents",
"split_indices", "split_indices",
"split_conditions", "split_conditions",
"default_left" "default_left",
"categories",
"categories_nodes",
"categories_segments",
"categories_sizes"
] ]
} }
}, },