ajitrajasekharan commited on
Commit
1d63fe1
1 Parent(s): 4b95e30

Update aggregate_server_json.py

Browse files
Files changed (1) hide show
  1. aggregate_server_json.py +2 -1
aggregate_server_json.py CHANGED
@@ -210,7 +210,7 @@ class AggregateNER:
210
  #print("mean",mean,"std_dev",std_dev)
211
  #threshold = mean_score - std_dev*self.threshold #default is 1 standard deviation from mean
212
  threshold = mean_score
213
- pick_count = 2
214
  for node in dist:
215
  print("conf:",node["confidence"],"threshold:",threshold)
216
  if (node["confidence"] >= threshold):
@@ -274,6 +274,7 @@ class AggregateNER:
274
  ret_arr = self.get_predictions_above_threshold(results[server_index]["orig_cs_prediction_details"][pivot_index])
275
  orig_cs_second_entity = results[server_index]["orig_cs_prediction_details"][pivot_index]['cs_distribution'][1]
276
  m2_cs = orig_cs_second_entity["e"].split('[')[0]
 
277
  is_cs_included = True if (m2_cs in servers_arr[server_index]["precedence"]) else False
278
  is_cs_included = True #Disabling cs included check. If prediction above threshold is cross prediction, then letting it through
279
  assert (m2_cs != m1)
 
210
  #print("mean",mean,"std_dev",std_dev)
211
  #threshold = mean_score - std_dev*self.threshold #default is 1 standard deviation from mean
212
  threshold = mean_score
213
+ pick_count = 1
214
  for node in dist:
215
  print("conf:",node["confidence"],"threshold:",threshold)
216
  if (node["confidence"] >= threshold):
 
274
  ret_arr = self.get_predictions_above_threshold(results[server_index]["orig_cs_prediction_details"][pivot_index])
275
  orig_cs_second_entity = results[server_index]["orig_cs_prediction_details"][pivot_index]['cs_distribution'][1]
276
  m2_cs = orig_cs_second_entity["e"].split('[')[0]
277
+ print("m2_cs",m2_cs,"ret_arr",ret_arr)
278
  is_cs_included = True if (m2_cs in servers_arr[server_index]["precedence"]) else False
279
  is_cs_included = True #Disabling cs included check. If prediction above threshold is cross prediction, then letting it through
280
  assert (m2_cs != m1)