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,
)
class Config(NewTask.Config):
model: RoBERTaWordTaggingModel.Config = RoBERTaWordTaggingModel.Config(
)
metric_reporter: NERMetricReporter.Config = NERMetricReporter.Config()
