qurator-spk/sbb_ner
1
0
Fork 0
mirror of https://github.com/qurator-spk/sbb_ner.git synced 2025-06-07 19:35:14 +02:00
Code Issues Releases Wiki Activity

fix model output handling bug

Browse source
This commit is contained in:
Kai Labusch 2020-05-19 16:57:53 +02:00
parent 319c29fc96
commit fab5d460f2
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
qurator/sbb_ner/webapp/app.py
View file

@ -244,6 +244,9 @@ def ner(model_id=None):
if not token.startswith('##') and word_pred == 'X': if not token.startswith('##') and word_pred == 'X':
word_pred = 'O' word_pred = 'O'
if word_pred == '[SEP]':
word_pred = 'O'
token = token[2:] if token.startswith('##') else token token = token[2:] if token.startswith('##') else token
word += token word += token