Commit
·
c839604
1
Parent(s):
193cf48
added 99.7 recall
Browse files
app.py
CHANGED
|
@@ -177,8 +177,8 @@ class App(object):
|
|
| 177 |
@staticmethod
|
| 178 |
def _get_threshold_map():
|
| 179 |
return {
|
| 180 |
-
'obi/deid_bert_i2b2':{"99.5": 4.656325975101986e-06},
|
| 181 |
-
'obi/deid_roberta_i2b2':{"99.5": 2.4362972672812125e-05}
|
| 182 |
}
|
| 183 |
|
| 184 |
@staticmethod
|
|
@@ -236,7 +236,7 @@ def deid(text, model, threshold):
|
|
| 236 |
deid_text_remove = list(app.get_deid_text_removed(tmp.name, tmp_1.name))[0]['deid_text']
|
| 237 |
return [highlight_text for highlight_text in App._get_highlights(deid_text)], deid_text_remove
|
| 238 |
|
| 239 |
-
recall_choices = ["No threshold", "99.5"]
|
| 240 |
recall_radio_input = gr.inputs.Radio(recall_choices, type="value", default=None, label='RECALL THRESHOLD')
|
| 241 |
|
| 242 |
model_choices = list(App._get_model_map().keys())
|
|
|
|
| 177 |
@staticmethod
|
| 178 |
def _get_threshold_map():
|
| 179 |
return {
|
| 180 |
+
'obi/deid_bert_i2b2':{"99.5": 4.656325975101986e-06, "99.7":1.8982457699258832e-06},
|
| 181 |
+
'obi/deid_roberta_i2b2':{"99.5": 2.4362972672812125e-05, "99.7":2.396420546444644e-06}
|
| 182 |
}
|
| 183 |
|
| 184 |
@staticmethod
|
|
|
|
| 236 |
deid_text_remove = list(app.get_deid_text_removed(tmp.name, tmp_1.name))[0]['deid_text']
|
| 237 |
return [highlight_text for highlight_text in App._get_highlights(deid_text)], deid_text_remove
|
| 238 |
|
| 239 |
+
recall_choices = ["No threshold", "99.5", "99.7"]
|
| 240 |
recall_radio_input = gr.inputs.Radio(recall_choices, type="value", default=None, label='RECALL THRESHOLD')
|
| 241 |
|
| 242 |
model_choices = list(App._get_model_map().keys())
|