adding thresholds and QE scores.

browsermt · Jun 25, 2020 · 92e0364 · 92e0364
1 parent cf7b5f7
commit 92e0364
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/src/service/quality_estimation/qe.cpp b/src/service/quality_estimation/qe.cpp
@@ -4,7 +4,16 @@ namespace qe {
 
 /// convert sentence score (log prob) to QE score
 int sentence_score2qe_score(const float sentence_score) {
-  return 42;
+  // empirically defined thresholds
+  float thresholds[2] = {-0.24869322, -0.10945034};
+
+  if (sentence_score < thresholds[0]) {
+    return 3;
+  } else if (sentence_score > thresholds[1]) {
+    return 1;
+  } else {
+    return 2;
+  }
 }
 
 }} // end of namespace marian::qea