Try to fix textcat test

This commit is contained in:
Matthew Honnibal 2020-07-09 20:24:53 +02:00
parent 018319a640
commit eb064c59cd
1 changed files with 1 additions and 1 deletions

View File

@ -84,7 +84,7 @@ def test_overfitting_IO():
# Simple test to try and quickly overfit the textcat component - ensuring the ML models work correctly # Simple test to try and quickly overfit the textcat component - ensuring the ML models work correctly
fix_random_seed(0) fix_random_seed(0)
nlp = English() nlp = English()
textcat = nlp.create_pipe("textcat") textcat = nlp.create_pipe("textcat", config={"exclusive_classes": True})
train_examples = [] train_examples = []
for text, annotations in TRAIN_DATA: for text, annotations in TRAIN_DATA:
train_examples.append(Example.from_dict(nlp.make_doc(text), annotations)) train_examples.append(Example.from_dict(nlp.make_doc(text), annotations))