Merge branch 'master' into spacy.io

This commit is contained in:
Ines Montani 2019-03-22 18:21:16 +01:00
commit 7b5496027b
2 changed files with 5 additions and 3 deletions

View File

@ -73,10 +73,12 @@ def main(model=None, output_dir=None, n_iter=20, n_texts=2000, init_tok2vec=None
textcat.model.tok2vec.from_bytes(file_.read())
print("Training the model...")
print("{:^5}\t{:^5}\t{:^5}\t{:^5}".format("LOSS", "P", "R", "F"))
batch_sizes = compounding(4.0, 32.0, 1.001)
for i in range(n_iter):
losses = {}
# batch up the examples using spaCy's minibatch
batches = minibatch(train_data, size=compounding(4.0, 32.0, 1.001))
random.shuffle(train_data)
batches = minibatch(train_data, size=batch_sizes)
for batch in batches:
texts, annotations = zip(*batch)
nlp.update(texts, annotations, sgd=optimizer, drop=0.2, losses=losses)

View File

@ -28,8 +28,8 @@
},
"spacyVersion": "2.1",
"binderUrl": "ines/spacy-io-binder",
"binderBranch": "nightly",
"binderVersion": "2.1.0a9",
"binderBranch": "live",
"binderVersion": "2.1.2",
"sections": [
{ "id": "usage", "title": "Usage Documentation", "theme": "blue" },
{ "id": "models", "title": "Models Documentation", "theme": "blue" },