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

Merge branch 'master' of https://github.com/tqchen/xgboost

Browse Source
This commit is contained in:
kalenhaha 2014-03-26 23:50:56 +08:00
commit 81b32525e0
4 changed files with 15 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
README.md
View File

@ -1,6 +1,6 @@
xgboost: eXtreme Gradient Boosting
=======
An efficient general purpose gradient boosting (tree) library.
A General purpose gradient boosting (tree) library.
Creater: Tianqi Chen
@ -32,8 +32,9 @@ Planned components
- matrix factorization
- structured prediction
File extension convention:
(1) .h are interface, utils and data structures, with detailed comment;
(2) .cpp are implementations that will be compiled, with less comment;
(3) .hpp are implementations that will be included by .cpp, with less comment
File extension convention
=======
* .h are interface, utils and data structures, with detailed comment;
* .cpp are implementations that will be compiled, with less comment;
* .hpp are implementations that will be included by .cpp, with less comment

2
demo/binary_classification/README
View File

@ -9,3 +9,5 @@ Format of featmap.txt:
q means continuous quantities, i means indicator features.
Feature id must be from 0 to num_features, in sorted order.
Detailed explaination: https://github.com/tqchen/xgboost/wiki/Binary-Classification

9
demo/binary_classification/runexp.sh
View File

@ -6,9 +6,10 @@ python mknfold.py agaricus.txt 1
# training and output the models
../../xgboost mushroom.conf
# output prediction task=pred
../../xgboost mushroom.conf task=pred model_in=0003.model
# print the boosters of 00003.model in dump.raw.txt
../../xgboost mushroom.conf task=dump model_in=0003.model name_dump=dump.raw.txt
../../xgboost mushroom.conf task=pred model_in=0002.model
# print the boosters of 00002.model in dump.raw.txt
../../xgboost mushroom.conf task=dump model_in=0002.model name_dump=dump.raw.txt
# use the feature map in printing for better visualization
../../xgboost mushroom.conf task=dump model_in=0003.model fmap=featmap.txt name_dump=dump.nice.txt
../../xgboost mushroom.conf task=dump model_in=0002.model fmap=featmap.txt name_dump=dump.nice.txt
cat dump.nice.txt

4
regression/xgboost_reg_main.cpp
View File

@ -111,7 +111,7 @@ namespace xgboost{
inline void InitData( void ){
if( name_fmap != "NULL" ) fmap.LoadText( name_fmap.c_str() );
if( task == "dump" ) return;
if( task == "test" || task == "dumppath" ){
if( task == "pred" || task == "dumppath" ){
data.CacheLoad( test_path.c_str(), silent!=0, use_buffer!=0 );
}else{
// training
@ -155,7 +155,7 @@ namespace xgboost{
// always save final round
if( save_period == 0 || num_round % save_period != 0 ){
if( model_out == "NULL" ){
this->SaveModel( num_round );
this->SaveModel( num_round - 1 );
}else{
this->SaveModel( model_out.c_str() );
}