ditto for merge prob

gucorpling · Jun 22, 2022 · d7ba27b · d7ba27b
1 parent 182568c
commit d7ba27b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/js/new_segmenter.js b/src/js/new_segmenter.js
@@ -40,8 +40,8 @@ export function Document(props) {
     let className = "sentence"
     const tokens = props.tokens.filter(t => t["token-type"] !== "super");
     const firstToken = tokens && tokens.length > 0 && tokens[0]
-    const outProb = firstToken && firstToken.probas && firstToken.probas.O
-    const maybeMerge = outProb && outProb > SSPLIT_SUSPICIOUS_PROBABILITY_THRESHOLD
+    const beginProb = firstToken && firstToken.probas && firstToken.probas.B
+    const maybeMerge = beginProb && beginProb < SSPLIT_SUSPICIOUS_PROBABILITY_THRESHOLD
     const isGold = firstToken.quality && firstToken.quality === "gold"
     if (!isGold && maybeMerge) {
       className += " sentence-maybe-merge"