pedramyazdipoor
commited on
Commit
•
9533e9d
1
Parent(s):
26c7565
Update README.md
Browse files
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 |
-
|
67 |
-
|
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 |
|