use SUSPICIOUS_PROBABILITY_THRESHOLD instead of 0.9

gucorpling · Jun 15, 2022 · de40ac9 · de40ac9
1 parent 7aec612
commit de40ac9
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 5 deletions.
diff --git a/src/js/new_segmenter.js b/src/js/new_segmenter.js
@@ -41,7 +41,7 @@ export function Document(props) {
     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 > 0.9
+    const maybeMerge = outProb && outProb > SUSPICIOUS_PROBABILITY_THRESHOLD
     const isGold = firstToken.quality && firstToken.quality === "gold"
     if (!isGold && maybeMerge) {
       className += " sentence-maybe-merge"

diff --git a/src/js/segmenter.js b/src/js/segmenter.js
@@ -101,7 +101,7 @@ function parse_input(conllu) {
 				// setting spacer shape and color
 				if (split_just_seen) {
 					space = "✗";
-					if (probas_num < 0.9) {
+					if (probas_num < SUSPICIOUS_PROBABILITY_THRESHOLD) {
 						testclass = "red";
 					} else{
 						testclass = "gray";

diff --git a/src/ud-tree/ud-tree/components.js b/src/ud-tree/ud-tree/components.js
@@ -271,13 +271,12 @@ function computeEdge (key, x, y, dx, dy, maxHeight, color, highlighted=false) {
 // Data/API helpers
 //////////////////////////////////////////////////////////////////////////////////
 // active learning functions for determiniting which annotations should be highlighted
-const SUSPICIOUS_CUTOFF = 0.9
 function isHeadSuspicious(head) {
-  return head.quality !== "gold" && head.probas && head.probas[head.value] < SUSPICIOUS_CUTOFF;
+  return head.quality !== "gold" && head.probas && head.probas[head.value] < SUSPICIOUS_PROBABILITY_THRESHOLD;
 }
 
 function isXposSuspicious(xpos) {
-  return xpos.quality !== "gold" && xpos.probas && xpos.probas[xpos.value] < SUSPICIOUS_CUTOFF;
+  return xpos.quality !== "gold" && xpos.probas && xpos.probas[xpos.value] < SUSPICIOUS_PROBABILITY_THRESHOLD;
 }
 
 async function updateHead(id, head){

diff --git a/webpack.dev.js b/webpack.dev.js
@@ -70,6 +70,7 @@ module.exports = {
   plugins: [
     new webpack.DefinePlugin({
       API_ENDPOINT: JSON.stringify("http://localhost:3000/api"),
+      SUSPICIOUS_PROBABILITY_THRESHOLD: 0.9,
     }),
     new HtmlWebpackPlugin({
       template: './src/page-document/tmpl.html',

diff --git a/webpack.prod.js b/webpack.prod.js
@@ -79,6 +79,7 @@ module.exports = {
   plugins: [
     new webpack.DefinePlugin({
       API_ENDPOINT: JSON.stringify("http://localhost:3000/api"),
+      SUSPICIOUS_PROBABILITY_THRESHOLD: 0.9,
     }),
     new HtmlWebpackPlugin({
       template: './src/page-document/tmpl.html',