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

Add JSON dump functionality documentation (#3600)

Browse Source
This commit is contained in:
Grace Lam 2018-08-16 16:32:04 -07:00 committed by Philip Cho
parent 324f3b5259
commit dd1fda449c
No known key found for this signature in database
GPG Key ID: A758FA046E1F6BB8
2 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
demo/binary_classification/README.md
View File

@ -102,7 +102,7 @@ After training, we can use the output model to get the prediction of the test da
For binary classification, the output predictions are probability confidence scores in [0,1], corresponds to the probability of the label to be positive.
#### Dump Model
This is a preliminary feature, so far only tree model support text dump. XGBoost can display the tree models in text files and we can scan the model in an easy way:
This is a preliminary feature, so only tree models support text dump. XGBoost can display the tree models in text or JSON files, and we can scan the model in an easy way:
```
../../xgboost mushroom.conf task=dump model_in=0002.model name_dump=dump.raw.txt
../../xgboost mushroom.conf task=dump model_in=0002.model fmap=featmap.txt name_dump=dump.nice.txt

4
doc/parameter.rst
View File

@ -361,6 +361,10 @@ The following parameters are only used in the console version of XGBoost
- Feature map, used for dumping model
* ``dump_format`` [default= ``text``] options: ``text``, ``json``
- Format of model dump file
* ``name_dump`` [default= ``dump.txt``]
- Name of model dump file