Gosse Minnema commited on
Commit
e366dda
1 Parent(s): f87bc2c

Fix bug in confidence thresholding

Browse files
Files changed (1) hide show
  1. sociolome/lome_webserver.py +1 -1
sociolome/lome_webserver.py CHANGED
@@ -70,7 +70,7 @@ def make_prediction(sentence, spacy_model, predictor, confidence_threshold):
70
  "roles": [
71
  {"boundary": bnd, "label": label}
72
  for bnd, label, probas in zip(arg_spans, arg_labels, label_probas)
73
- if label != "Target" and max(probas) < confidence_threshold
74
  ]
75
  }
76
 
 
70
  "roles": [
71
  {"boundary": bnd, "label": label}
72
  for bnd, label, probas in zip(arg_spans, arg_labels, label_probas)
73
+ if label != "Target" and max(probas) > confidence_threshold
74
  ]
75
  }
76