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

fix softmax

Browse Source
This commit is contained in:
tqchen 2014-05-16 20:28:07 -07:00
parent 6c72d02205
commit b07ff1ac8d
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
regrank/xgboost_regrank.h
View File

@ -97,8 +97,8 @@ namespace xgboost{
*/ */
inline void InitTrainer(void){ inline void InitTrainer(void){
if( mparam.num_class != 0 ){ if( mparam.num_class != 0 ){
if( name_obj_ != "softmax" ){ if( name_obj_ != "multi:softmax" ){
name_obj_ = "softmax"; name_obj_ = "multi:softmax";
printf("auto select objective=softmax to support multi-class classification\n" ); printf("auto select objective=softmax to support multi-class classification\n" );
} }
} }