ajitrajasekharan commited on
Commit
f8e99fa
1 Parent(s): 0ebf7a3

Update aggregate_server_json.py

Browse files
Files changed (1) hide show
  1. aggregate_server_json.py +1 -1
aggregate_server_json.py CHANGED
@@ -289,7 +289,7 @@ class AggregateNER:
289
  else:
290
  return flip_category(results[server_index]["ner"][run_index])
291
  else:
292
- #here cs and ci are same. So use two cs predictions if meaningful
293
  if (len(results[server_index]["orig_cs_prediction_details"][pivot_index]['cs_distribution']) >= 2):
294
  ret_arr = self.get_predictions_above_threshold(results[server_index]["orig_cs_prediction_details"][pivot_index])
295
  orig_cs_second_entity = results[server_index]["orig_cs_prediction_details"][pivot_index]['cs_distribution'][1]
 
289
  else:
290
  return flip_category(results[server_index]["ner"][run_index])
291
  else:
292
+ #here cs and ci are same. So use two consecutive cs predictions if meaningful
293
  if (len(results[server_index]["orig_cs_prediction_details"][pivot_index]['cs_distribution']) >= 2):
294
  ret_arr = self.get_predictions_above_threshold(results[server_index]["orig_cs_prediction_details"][pivot_index])
295
  orig_cs_second_entity = results[server_index]["orig_cs_prediction_details"][pivot_index]['cs_distribution'][1]