small fix

This commit is contained in:
tqchen
2014-03-25 17:17:00 -07:00
parent f7d9c774d7
commit c7869a7855
2 changed files with 6 additions and 5 deletions

View File

@@ -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() );
}