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