Get the n-best logits from a list.
(logits, n_best_size)
| 210 | |
| 211 | |
| 212 | def _get_best_indexes(logits, n_best_size): |
| 213 | """Get the n-best logits from a list.""" |
| 214 | |
| 215 | index_and_score = sorted(enumerate(logits), key=lambda x: x[1], reverse=True) |
| 216 | |
| 217 | best_indexes = [] |
| 218 | for i in range(len(index_and_score)): |
| 219 | if i >= n_best_size: |
| 220 | break |
| 221 | best_indexes.append(index_and_score[i][0]) |
| 222 | return best_indexes |
| 223 | |
| 224 | |
| 225 | def get_final_text(pred_text, orig_text, do_lower_case): |
no test coverage detected