change rank order output to follow kaggle convention

2014-05-25 10:08:38 -07:00 · 2014-05-25 10:08:38 -07:00 · bf5fcec8e8
commit bf5fcec8e8
parent 278b788b34
1 changed files with 2 additions and 1 deletions
--- 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()