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

change rank order output to follow kaggle convention

Browse Source
This commit is contained in:
Tianqi Chen 2014-05-25 10:08:38 -07:00
parent 278b788b34
commit bf5fcec8e8
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
demo/kaggle-higgs/higgs-pred.py
View File

@ -43,7 +43,8 @@ for k, v in res:
nhit += 1 nhit += 1
else: else:
lb = 'b' lb = 'b'
fo.write('%s,%d,%s\n' % ( k, rorder[k], lb ) ) # change output rank order to follow Kaggle convention
fo.write('%s,%d,%s\n' % ( k, len(rorder)+1-rorder[k], lb ) )
ntot += 1 ntot += 1
fo.close() fo.close()