Esempio n. 1
0
 def create_metric_reporter(cls, config: Config,
                            tensorizers: Dict[str, Tensorizer]):
     return NERMetricReporter(
         channels=[ConsoleChannel()],
         label_names=list(tensorizers["tokens"].labels_vocab._vocab),
         pad_idx=tensorizers["tokens"].labels_pad_idx,
     )
Esempio n. 2
0
 class Config(NewTask.Config):
     model: RoBERTaWordTaggingModel.Config = RoBERTaWordTaggingModel.Config(
     )
     metric_reporter: NERMetricReporter.Config = NERMetricReporter.Config()