xgboost源代码

#!/usr/bin/python

def loadfmap（ fname ）:
    fmap = {}
    nmap = {}

    for l in open（ fname ）:
        arr = l.split（）
        if arr[0].find（‘.‘） != -1:
            idx = int（ arr[0].strip（‘.‘） ）
            assert idx not in fmap
            fmap[ idx ] = {}
            ftype = arr[1].strip（‘:‘）
            content = arr[2]
        else:
            content = arr[0]
        for it in content.split（‘‘）:
            if it.strip（） == ‘‘:
                continue
            k  v = it.split（‘=‘）
            fmap[ idx ][ v ] = len（nmap）
            nmap[ len（nmap） ] = ftype+‘=‘+k
    return fmap nmap

def write_nmap（ fo nmap ）:
    for i in range（ len（nmap） ）:
        fo.write（‘%d	%s	i
‘ % （i nmap[i]） ）

# start here
fmap nmap = loadfmap（ ‘agaricus-lepiota.fmap‘ ）
fo = open（ ‘featmap.txt‘ ‘w‘ ）
write_nmap（ fo nmap ）
fo.close（）

fo = open（ ‘agaricus.txt‘ ‘w‘ ）
for l in open（ ‘agaricus-lepiota.data‘ ）:
    arr = l.split（‘‘）
    if arr[0] == ‘p‘:
        fo.write（‘1‘）
    else:
        assert arr[0] == ‘e‘
        fo.write（‘0‘）
    for i in range（ 1len（arr） ）:
        fo.write（ ‘ %d:1‘ % fmap[i][arr[i].strip（）] ）
    fo.write（‘
‘）

fo.close（）

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----
     目录           0  2017-05-08 16:20  xgboost-master
     文件         681  2015-12-09 04:45  xgboost-master.gitignore
     文件        1064  2015-12-09 04:45  xgboost-master.travis.yml
     文件        2085  2015-12-09 04:45  xgboost-masterCHANGES.md
     文件        3443  2015-12-09 04:45  xgboost-masterCONTRIBUTORS.md
     文件         564  2015-12-09 04:45  xgboost-masterLICENSE
     文件        5920  2015-12-09 04:45  xgboost-masterMakefile
     目录           0  2017-05-08 16:20  xgboost-masterR-package
     文件          56  2015-12-09 04:45  xgboost-masterR-package.Rbuildignore
     文件        1340  2015-12-09 04:45  xgboost-masterR-packageDEscriptION
     文件         580  2015-12-09 04:45  xgboost-masterR-packageLICENSE
     文件        1239  2015-12-09 04:45  xgboost-masterR-packageNAMESPACE
     目录           0  2017-05-08 16:20  xgboost-masterR-packageR
     文件        1853  2015-12-09 04:45  xgboost-masterR-packageRgetinfo.xgb.DMatrix.R
     文件         530  2015-12-09 04:45  xgboost-masterR-packageR
row.xgb.DMatrix.R
     文件        3190  2015-12-09 04:45  xgboost-masterR-packageRpredict.xgb.Booster.R
     文件         553  2015-12-09 04:45  xgboost-masterR-packageRpredict.xgb.Booster.handle.R
     文件        1155  2015-12-09 04:45  xgboost-masterR-packageRsetinfo.xgb.DMatrix.R
     文件        1477  2015-12-09 04:45  xgboost-masterR-packageRslice.xgb.DMatrix.R
     文件       11951  2015-12-09 04:45  xgboost-masterR-packageRutils.R
     文件        1655  2015-12-09 04:45  xgboost-masterR-packageRxgb.DMatrix.R
     文件         795  2015-12-09 04:45  xgboost-masterR-packageRxgb.DMatrix.save.R
     文件        4198  2015-12-09 04:45  xgboost-masterR-packageRxgb.create.features.R
     文件       11244  2015-12-09 04:45  xgboost-masterR-packageRxgb.cv.R
     文件        2971  2015-12-09 04:45  xgboost-masterR-packageRxgb.dump.R
     文件        7437  2015-12-09 04:45  xgboost-masterR-packageRxgb.importance.R
     文件         976  2015-12-09 04:45  xgboost-masterR-packageRxgb.load.R
     文件        7324  2015-12-09 04:45  xgboost-masterR-packageRxgb.model.dt.tree.R
     文件        5572  2015-12-09 04:45  xgboost-masterR-packageRxgb.plot.deepness.R
     文件        3976  2015-12-09 04:45  xgboost-masterR-packageRxgb.plot.importance.R
     文件        5435  2015-12-09 04:45  xgboost-masterR-packageRxgb.plot.multi.trees.R
............此处省略560个文件信息