pedramyazdipoor commited on
Commit
9533e9d
1 Parent(s): 26c7565

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -4
README.md CHANGED
@@ -63,10 +63,8 @@ def generate_indexes(start_logits, end_logits, N, min_index_list):
63
  end_probs = output_end
64
  list_end = dict(zip(end_indexes, end_probs.tolist()))
65
 
66
- sort_start_probs = sorted(list_start.items(), key=lambda x: x[1], reverse=True) #Descending sort by probability
67
- sort_end_probs = sorted(list_end.items(), key=lambda x: x[1], reverse=True)
68
- sorted_start_list = (sort_start_probs)
69
- sorted_end_list = (sort_end_probs)
70
 
71
  final_start_idx, final_end_idx = [[] for l in range(2)]
72
 
 
63
  end_probs = output_end
64
  list_end = dict(zip(end_indexes, end_probs.tolist()))
65
 
66
+ sorted_start_list = sorted(list_start.items(), key=lambda x: x[1], reverse=True) #Descending sort by probability
67
+ sorted_end_list = sorted(list_end.items(), key=lambda x: x[1], reverse=True)
 
 
68
 
69
  final_start_idx, final_end_idx = [[] for l in range(2)]
70