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

demo

Browse Source
This commit is contained in:
antinucleon 2014-05-16 21:05:11 -06:00
parent 615074efb6
commit 2fcd875675
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
demo/multi_classification/train.py
View File

@ -1,4 +1,4 @@
#! /usr/bin/python
import sys import sys
import numpy as np import numpy as np
sys.path.append('../../python/') sys.path.append('../../python/')
@ -29,7 +29,6 @@ param['objective'] = 'multi:softmax'
# scale weight of positive examples # scale weight of positive examples
param['bst:eta'] = 0.1 param['bst:eta'] = 0.1
param['bst:max_depth'] = 6 param['bst:max_depth'] = 6
param['eval_metric'] = 'auc'
param['silent'] = 1 param['silent'] = 1
param['nthread'] = 4 param['nthread'] = 4
param['num_class'] = 5 param['num_class'] = 5