pierreguillou commited on
Commit
ed945a4
1 Parent(s): b24b127

Update files/functions.py

Browse files
Files changed (1) hide show
  1. files/functions.py +1 -0
files/functions.py CHANGED
@@ -740,6 +740,7 @@ def predictions_paragraph_level(dataset, outputs, images_ids_list, chunk_ids, in
740
  probs_label = list()
741
  for probs_list in probs:
742
  prob_label = reduce(lambda x, y: x*y, probs_list)
 
743
  probs_label.append(prob_label)
744
  max_value = max(probs_label)
745
  max_index = probs_label.index(max_value)
 
740
  probs_label = list()
741
  for probs_list in probs:
742
  prob_label = reduce(lambda x, y: x*y, probs_list)
743
+ prob_label = prob_label**(1./(len(probs_list))) # normalization
744
  probs_label.append(prob_label)
745
  max_value = max(probs_label)
746
  max_index = probs_label.index(max_value)