This commit is contained in:
Matthew Honnibal 2017-04-24 00:25:55 +02:00
parent 70a43858e1
commit 65f10b53e5
1 changed files with 3 additions and 1 deletions

View File

@ -56,6 +56,7 @@ def test_issue999(train_data):
for _, offsets in train_data: for _, offsets in train_data:
for start, end, ent_type in offsets: for start, end, ent_type in offsets:
nlp.entity.add_label(ent_type) nlp.entity.add_label(ent_type)
nlp.entity.model.learn_rate = 0.001
for itn in range(100): for itn in range(100):
random.shuffle(train_data) random.shuffle(train_data)
for raw_text, entity_offsets in train_data: for raw_text, entity_offsets in train_data:
@ -74,4 +75,5 @@ def test_issue999(train_data):
assert ents[(start, end)] == label assert ents[(start, end)] == label
break break
else: else:
if entity_offsets:
raise Exception(ents) raise Exception(ents)