Prechádzať zdrojové kódy

fix test-label bug

Summary: `Meter` didn't count golds for individual numbers any more when `falseNegativeLabels_` is false.

Reviewed By: EdouardGrave

Differential Revision: D21379435

fbshipit-source-id: b898c5d156727a0f001db5efbf72fd0bf6640efb
Onur Çelebi 5 rokov pred
rodič
commit
b64e359d54
1 zmenil súbory, kde vykonal 3 pridanie a 3 odobranie
  1. 3 3
      src/meter.cc

+ 3 - 3
src/meter.cc

@@ -39,9 +39,9 @@ void Meter::log(
     labelMetrics_[prediction.second].scoreVsTrue.emplace_back(score, gold);
   }
 
-  if (falseNegativeLabels_) {
-    for (const auto& label : labels) {
-      labelMetrics_[label].gold++;
+  for (const auto& label : labels) {
+    labelMetrics_[label].gold++;
+    if (falseNegativeLabels_) {
       if (!utils::containsSecond(predictions, label)) {
         labelMetrics_[label].scoreVsTrue.emplace_back(falseNegativeScore, 1.0);
       }