From bf5fcec8e8322e67b5f25a913a26c9d8c1de1d28 Mon Sep 17 00:00:00 2001 From: Tianqi Chen Date: Sun, 25 May 2014 10:08:38 -0700 Subject: [PATCH] change rank order output to follow kaggle convention --- demo/kaggle-higgs/higgs-pred.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/demo/kaggle-higgs/higgs-pred.py b/demo/kaggle-higgs/higgs-pred.py index a38f6d82a..18ba711f0 100755 --- a/demo/kaggle-higgs/higgs-pred.py +++ b/demo/kaggle-higgs/higgs-pred.py @@ -43,7 +43,8 @@ for k, v in res: nhit += 1 else: 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 fo.close()