Spaces:
Runtime error
Runtime error
Commit
·
4cf6282
1
Parent(s):
4e4285d
Update aggregate_server_json.py
Browse files- aggregate_server_json.py +2 -1
aggregate_server_json.py
CHANGED
|
@@ -295,11 +295,12 @@ class AggregateNER:
|
|
| 295 |
print("***** here 2")
|
| 296 |
if (len(results[server_index]["orig_cs_prediction_details"][pivot_index]['cs_distribution']) >= 2):
|
| 297 |
ret_arr = self.get_predictions_above_threshold(results[server_index]["orig_cs_prediction_details"][pivot_index])
|
|
|
|
| 298 |
orig_cs_second_entity = results[server_index]["orig_cs_prediction_details"][pivot_index]['cs_distribution'][1]
|
| 299 |
m2_cs = orig_cs_second_entity["e"].split('[')[0]
|
| 300 |
print("m2_cs",m2_cs,"ret_arr",ret_arr)
|
| 301 |
is_cs_included = True if (m2_cs in servers_arr[server_index]["precedence"]) else False
|
| 302 |
-
|
| 303 |
assert (m2_cs != m1)
|
| 304 |
if (is_cs_included and self.check_if_entity_in_arr(m2_cs,ret_arr)):
|
| 305 |
ret_obj = results[server_index]["ner"][run_index].copy()
|
|
|
|
| 295 |
print("***** here 2")
|
| 296 |
if (len(results[server_index]["orig_cs_prediction_details"][pivot_index]['cs_distribution']) >= 2):
|
| 297 |
ret_arr = self.get_predictions_above_threshold(results[server_index]["orig_cs_prediction_details"][pivot_index])
|
| 298 |
+
print("orig cs:",results[server_index]["orig_cs_prediction_details"][pivot_index]['cs_distribution'])
|
| 299 |
orig_cs_second_entity = results[server_index]["orig_cs_prediction_details"][pivot_index]['cs_distribution'][1]
|
| 300 |
m2_cs = orig_cs_second_entity["e"].split('[')[0]
|
| 301 |
print("m2_cs",m2_cs,"ret_arr",ret_arr)
|
| 302 |
is_cs_included = True if (m2_cs in servers_arr[server_index]["precedence"]) else False
|
| 303 |
+
is_cs_included = True #Disabling cs included check. If prediction above threshold is cross prediction, then letting it through
|
| 304 |
assert (m2_cs != m1)
|
| 305 |
if (is_cs_included and self.check_if_entity_in_arr(m2_cs,ret_arr)):
|
| 306 |
ret_obj = results[server_index]["ner"][run_index].copy()
|