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

add rank

Browse Source
This commit is contained in:
tqchen 2014-04-30 09:32:42 -07:00
parent a383f11759
commit bb93c0aaac
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
regrank/xgboost_regrank_obj.hpp
View File

@ -81,6 +81,7 @@ namespace xgboost{
rec.clear(); rec.clear();
for(unsigned j = gptr[k]; j < gptr[k+1]; ++j ){ for(unsigned j = gptr[k]; j < gptr[k+1]; ++j ){
rec.push_back( std::make_pair(info.labels[j], j) ); rec.push_back( std::make_pair(info.labels[j], j) );
grad[j] = hess[j] = 0.0f;
} }
std::sort( rec.begin(), rec.end(), CmpFirst ); std::sort( rec.begin(), rec.end(), CmpFirst );
// enumerate buckets with same label, for each item in the list, grab another sample randomly // enumerate buckets with same label, for each item in the list, grab another sample randomly