1
0
Fork 0
mirror of https://github.com/qurator-spk/sbb_ner.git synced 2025-06-09 04:10:08 +02:00

permit empty input

This commit is contained in:
Kai Labusch 2019-12-02 15:53:16 +01:00
parent 5f6d045755
commit 5d2a75b2b2

View file

@ -119,7 +119,13 @@ class NERPredictor:
prev_guid = fe.guid
assert len(sentences) == len(prediction)
try:
assert len(sentences) == len(prediction)
except AssertionError:
print('Sentences:\n')
print(sentences)
print('\n\nPrediciton:\n')
print(prediction)
return prediction
@ -142,6 +148,9 @@ class NERTokenizer:
sen = [tok.replace(" ", "") for tok in sen]
if len(sen) == 0:
continue
sentences.append((sen, []))
return sentences