spaCy/spacy/tests/regression/test_issue4348.py

24 lines
687 B
Python
Raw Normal View History

# coding: utf8
from __future__ import unicode_literals
from spacy.lang.en import English
from spacy.util import minibatch, compounding
def test_issue4348():
"""Test that training the tagger with empty data, doesn't throw errors"""
TRAIN_DATA = [("", {"tags": []}), ("", {"tags": []})]
nlp = English()
tagger = nlp.create_pipe("tagger")
nlp.add_pipe(tagger)
optimizer = nlp.begin_training()
for i in range(5):
losses = {}
batches = minibatch(TRAIN_DATA, size=compounding(4.0, 32.0, 1.001))
for batch in batches:
texts, annotations = zip(*batch)
nlp.update(texts, annotations, sgd=optimizer, losses=losses)