Pass smaller doc sample into model initialize

This commit is contained in:
Matthw Honnibal 2020-05-21 20:17:24 +02:00
parent 3b5cfec1fc
commit df87c32a40

View File

@ -624,9 +624,10 @@ cdef class Parser:
sgd = self.create_optimizer() sgd = self.create_optimizer()
doc_sample = [] doc_sample = []
gold_sample = [] gold_sample = []
for example in islice(get_examples(), 1000): for example in islice(get_examples(), 10):
parses = example.get_gold_parses(merge=False, vocab=self.vocab) parses = example.get_gold_parses(merge=False, vocab=self.vocab)
for doc, gold in parses: for doc, gold in parses:
if len(doc):
doc_sample.append(doc) doc_sample.append(doc)
gold_sample.append(gold) gold_sample.append(gold)
for name, component in pipeline: for name, component in pipeline: