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

fix mushroom

Browse Source
This commit is contained in:
tqchen 2014-02-24 23:19:58 -08:00
parent c4949c0937
commit 6fa5c30777
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
booster/tree/xgboost_tree_model.h
View File

@ -298,10 +298,10 @@ namespace xgboost{
}else{
// right then left,
TSplitCond cond = nodes[ nid ].split_cond();
fprintf( fo, "%d:[f%u>%f] yes=%d,no=%d\n", nid,
nodes[ nid ].split_index(), float(cond), nodes[ nid ].cright(), nodes[ nid ].cleft() );
this->Dump( nodes[ nid ].cright(), fo, depth+1 );
fprintf( fo, "%d:[f%u<%f] yes=%d,no=%d\n", nid,
nodes[ nid ].split_index(), float(cond), nodes[ nid ].cleft(), nodes[ nid ].cright() );
this->Dump( nodes[ nid ].cleft(), fo, depth+1 );
this->Dump( nodes[ nid ].cright(), fo, depth+1 );
}
}
};

2
demo/mushroom/maptree.py
View File

@ -15,7 +15,7 @@ for l in open( 'dump.txt'):
if idx == -1:
fo.write(l)
else:
fid = int( l[idx+2:len(l)].split('>')[0])
fid = int( l[idx+2:len(l)].split('<')[0])
rl = l[0:idx]+'['+nmap[fid]+']' + l.split()[1].strip()+'\n'
fo.write(rl)