This commit is contained in:
ines 2017-06-04 22:36:40 +02:00
parent f432bb4b48
commit 96867a24ae
1 changed files with 2 additions and 1 deletions

View File

@ -79,7 +79,8 @@ def test_issue910(EN, train_data, additional_entity_types):
2) There's no way to set the learning rate for the weight update, so we 2) There's no way to set the learning rate for the weight update, so we
end up out-of-scale, causing it to learn too fast. end up out-of-scale, causing it to learn too fast.
''' '''
doc = EN(u"I am looking for a restaurant in Berlin") nlp = EN
doc = nlp(u"I am looking for a restaurant in Berlin")
ents_before_train = [(ent.label_, ent.text) for ent in doc.ents] ents_before_train = [(ent.label_, ent.text) for ent in doc.ents]
# Fine tune the ner model # Fine tune the ner model
for entity_type in additional_entity_types: for entity_type in additional_entity_types: