for BIO tag
(tags)
| 8 | |
| 9 | |
| 10 | def get_spans(tags): |
| 11 | '''for BIO tag''' |
| 12 | tags = tags.strip().split() |
| 13 | length = len(tags) |
| 14 | spans = [] |
| 15 | start = -1 |
| 16 | for i in range(length): |
| 17 | if tags[i].endswith('B'): |
| 18 | if start != -1: |
| 19 | spans.append([start, i - 1]) |
| 20 | start = i |
| 21 | elif tags[i].endswith('O'): |
| 22 | if start != -1: |
| 23 | spans.append([start, i - 1]) |
| 24 | start = -1 |
| 25 | if start != -1: |
| 26 | spans.append([start, length - 1]) |
| 27 | return spans |
| 28 | |
| 29 | |
| 30 | def get_evaluate_spans(tags, length, token_range): |