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 23f4c41035
commit 416050d5c0
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){
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" );
}
}