support int type

2014-03-12 17:58:14 -07:00
parent 329cc61795
commit 8c8dd1a740
3 changed files with 23 additions and 5 deletions
--- a/regression/xgboost_reg.h
+++ b/regression/xgboost_reg.h
@@ -292,7 +292,7 @@ namespace xgboost{
                * \brief adjust base_score
                */                
                inline void AdjustBase( void ){
-                    if( loss_type == 1 ){
+                    if( loss_type == 1 || loss_type == 2 ){
                        utils::Assert( base_score > 0.0f && base_score < 1.0f, "sigmoid range constrain" );
                        base_score = - logf( 1.0f / base_score - 1.0f );
                    }