Commit 2f0fb204 authored by Guolin Ke's avatar Guolin Ke
Browse files

fix #816

parent 1a7d3aa9
...@@ -86,7 +86,7 @@ public: ...@@ -86,7 +86,7 @@ public:
data_size_t idx = sorted_idx[j]; data_size_t idx = sorted_idx[j];
if (label[idx] > 0.5f) { if (label[idx] > 0.5f) {
++num_hit; ++num_hit;
sum_ap += static_cast<double>(num_hit) / (i + 1.0f); sum_ap += static_cast<double>(num_hit) / (j + 1.0f);
} }
} }
(*out)[i] = sum_ap / cur_k; (*out)[i] = sum_ap / cur_k;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment