Exemple #1
0
 def __init__(self, config, name, tokenizer=None):
     super(Unsupervised,
           self).__init__(config, name,
                          LabeledDataLoader(config, name, tokenizer))
     self.config = config
Exemple #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
Exemple #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
Exemple #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()))
Exemple #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()))