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

chg python back

Browse Source
This commit is contained in:
tqchen 2014-08-23 20:53:56 -07:00
parent d16a56814b
commit fea7245fa0
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -53,7 +53,7 @@ plst = list(param.items())+[('eval_metric', 'ams@0.15')]
watchlist = [ (xgmat,'train') ] watchlist = [ (xgmat,'train') ]
# boost 120 tres # boost 120 tres
num_round = 2 num_round = 120
print ('loading data end, start to boost trees') print ('loading data end, start to boost trees')
bst = xgb.train( plst, xgmat, num_round, watchlist ); bst = xgb.train( plst, xgmat, num_round, watchlist );
# save out model # save out model