pierreguillou commited on
Commit
e700c4f
·
1 Parent(s): bbe88e1

Update files/functions.py

Browse files
Files changed (1) hide show
  1. files/functions.py +1 -0
files/functions.py CHANGED
@@ -686,6 +686,7 @@ def predictions_paragraph_level_gradio(dataset, outputs, images_ids_list, chunk_
686
  probs_label = list()
687
  for probs_list in probs:
688
  prob_label = reduce(lambda x, y: x*y, probs_list)
 
689
  probs_label.append(prob_label)
690
  max_value = max(probs_label)
691
  max_index = probs_label.index(max_value)
 
686
  probs_label = list()
687
  for probs_list in probs:
688
  prob_label = reduce(lambda x, y: x*y, probs_list)
689
+ prob_label = prob_label**(1./(len(probs_list))) # normalization
690
  probs_label.append(prob_label)
691
  max_value = max(probs_label)
692
  max_index = probs_label.index(max_value)