explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-10-02 18:06:46 +03:00
Code Issues Packages Projects Releases Wiki Activity

Update example

Browse Source
This commit is contained in:
ines 2017-11-06 02:40:44 +01:00
parent 68b7e3a33a
commit 47969e306b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/training/train_pytorch_textcat.py
View File

@ -40,7 +40,7 @@ def main(model=None, output_dir=None, n_iter=20, n_texts=2000):
# Create the PyTorch neural network model, and wrap it with Thinc. This
# gives it the API that spaCy expects.
pt_model = create_model()
textcat = thinc.extra.wrappers.PytorchWrapper(pt_model)
textcat = thinc.extra.wrappers.PyTorchWrapper(pt_model)
nlp.add_pipe(textcat, last=True)
# add label to text classifier