Spaces:
Sleeping
Sleeping
Gosse Minnema
commited on
Commit
•
e366dda
1
Parent(s):
f87bc2c
Fix bug in confidence thresholding
Browse files
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)
|
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 |
|