From b07ff1ac8d4625c7b59ac2ce59d503d1a1c9c4c8 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" ); } }