From 416050d5c0f8dd11c5e5b75f274b6cdafef83251 Mon Sep 17 00:00:00 2001 From: tqchen Date: Fri, 16 May 2014 20:28:07 -0700 Subject: [PATCH] fix softmax --- regrank/xgboost_regrank.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/regrank/xgboost_regrank.h b/regrank/xgboost_regrank.h index b2649735c..e3868002a 100644 --- a/regrank/xgboost_regrank.h +++ b/regrank/xgboost_regrank.h @@ -97,8 +97,8 @@ namespace xgboost{ */ inline void InitTrainer(void){ if( mparam.num_class != 0 ){ - if( name_obj_ != "softmax" ){ - name_obj_ = "softmax"; + if( name_obj_ != "multi:softmax" ){ + name_obj_ = "multi:softmax"; printf("auto select objective=softmax to support multi-class classification\n" ); } }