Пример #1
0
 def __init__(self, config, name, tokenizer=None):
     super(Unsupervised,
           self).__init__(config, name,
                          LabeledDataLoader(config, name, tokenizer))
     self.config = config
Пример #2
0
 def __init__(self, config, name, tokenizer=None):
   super(SpanExtraction, self).__init__(
     config, name, LabeledDataLoader(config, name, tokenizer))
   self.n_classes = 2
   self.config = config
Пример #3
0
 def __init__(self, config, name, tokenizer=None):
   super(ECPExtraction, self).__init__(
     config, name, LabeledDataLoader(config, name, tokenizer))
   self.n_classes = len(set(self.loader.label_mapping.values()))
   self.config = config
Пример #4
0
 def __init__(self, config, name, tokenizer=None):
   super(SpanGCN, self).__init__(
     config, name, LabeledDataLoader(config, name, tokenizer))
   self.span_n_classes = len(set(self.loader.label_mapping.values()))
Пример #5
0
 def __init__(self, config, name, tokenizer):
     super().__init__(config, name,
                      LabeledDataLoader(config, name, tokenizer))
     self.n_classes = len(set(self.loader.label_mapping.values()))