Fix number characters

This commit is contained in:
Matthw Honnibal 2019-10-24 17:34:16 +02:00 committed by Matthew Honnibal
parent 29f8413095
commit f597992411

View File

@ -848,7 +848,7 @@ class ClozeMultitask(Pipe):
@classmethod
def Model(cls, vocab, tok2vec, **cfg):
if cfg["objective"] == "characters":
out_sizes = [256] * cfg.get("nr_char", 10)
out_sizes = [256] * cfg.get("nr_char", 4)
output_layer = MultiSoftmax(out_sizes)
else:
output_size = vocab.vectors.data.shape[1]
@ -867,7 +867,7 @@ class ClozeMultitask(Pipe):
self.model = model
self.cfg = cfg
self.cfg.setdefault("objective", "characters")
self.cfg.setdefault("nr_char", 10)
self.cfg.setdefault("nr_char", 4)
def set_annotations(self, docs, dep_ids, tensors=None):
pass
@ -1094,6 +1094,7 @@ cdef class DependencyParser(Parser):
assigns = ["token.dep", "token.is_sent_start", "doc.sents"]
requires = []
TransitionSystem = ArcEager
nr_feature = 8
@property
def postprocesses(self):