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

fix regression

Browse Source
This commit is contained in:
tqchen
2014-11-16 11:38:21 -08:00
parent 02c2278f96
commit 129fee64f3
3 changed files with 22 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
demo/kaggle-higgs/higgs-cv.py
View File

@@ -10,7 +10,7 @@ label = train[:,32]
data = train[:,1:31]
weight = train[:,31]
dtrain = xgb.DMatrix( data, label=label, missing = -999.0, weight=weight )
param = {'max_depth':6, 'eta':0.1, 'silent':1, 'objective':'binary:logitraw', 'nthread':4}
param = {'max_depth':6, 'eta':0.1, 'silent':1, 'objective':'binary:logitraw', 'nthread':4, 'updater':'grow_histmaker,prune'}
num_round = 120
print ('running cross validation, with preprocessing function')