add rank

2014-04-30 09:32:42 -07:00 · 2014-04-30 09:32:42 -07:00 · bb93c0aaac
commit bb93c0aaac
parent a383f11759
1 changed files with 2 additions and 1 deletions
--- a/regrank/xgboost_regrank_obj.hpp
+++ b/regrank/xgboost_regrank_obj.hpp
@ -81,7 +81,8 @@ namespace xgboost{
                        rec.clear();
                        for(unsigned j = gptr[k]; j < gptr[k+1]; ++j ){
                            rec.push_back( std::make_pair(info.labels[j], j) );
-                        }
+                            grad[j] = hess[j] = 0.0f;
+                        }                        
                        std::sort( rec.begin(), rec.end(), CmpFirst );
                        // enumerate buckets with same label, for each item in the list, grab another sample randomly
                        for( unsigned i = 0; i < rec.size(); ){