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": {
"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"
]
}
},