def __init__(self, conf: MySRLConf, vocab_evt: SimpleVocab, vocab_arg: SimpleVocab): self.conf = conf self.vocab_evt = vocab_evt self.vocab_arg = vocab_arg # -- self.evt_span_getter = Mention.create_span_getter(conf.evt_span_mode) self.evt_span_setter = Mention.create_span_setter(conf.evt_span_mode) self.arg_span_getter = Mention.create_span_getter(conf.arg_span_mode) self.arg_span_setter = Mention.create_span_setter(conf.arg_span_mode)
def __init__(self, conf: ExtenderConf, **kwargs): super().__init__(conf, **kwargs) conf: ExtenderConf = self.conf # -- self.ext_span_getter = Mention.create_span_getter(conf.ext_span_mode) self.ext_span_setter = Mention.create_span_setter(conf.ext_span_mode) self.eenc = PlainEncoder(conf.eenc_conf, input_dim=conf.isize) self.enode = SpanExpanderNode(conf.econf, isize=conf.isize, psize=(conf.psize if conf.ext_use_finput else -1))
def __init__(self, conf: BaseExtractorConf, vocab: SimpleVocab): self.conf = conf # borrow conf self.core_span_getter = Mention.create_span_getter(conf.core_span_mode) self.core_span_setter = Mention.create_span_setter(conf.core_span_mode) self.vocab = vocab # -- if conf.ftag == "arg": self._get_f = self._get_args self._clear_f = lambda inst: inst.clear_args() # first delete all args if existing self._new_f = self._new_arg else: self._get_f = self._get_frames self._clear_f = lambda inst: inst.delete_frames(conf.ftag) # first delete all frames if existing self._new_f = self._new_frame