prajwal967
commited on
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())
|