def update_metrics(self, batch: Dict[str, Any], output: Union[torch.Tensor, Dict[str, torch.Tensor], Iterable[torch.Tensor], Any], prediction: Dict[str, Any], metric: Union[MetricDict, Metric]): BiaffineSemanticDependencyParser.update_metric(self, *prediction, batch['arc'], batch['rel_id'], output[1], output[-1], metric, batch)
def update_metric(self, arc_preds, rel_preds, arcs, rels, mask, puncts, metric, batch=None): super().update_metric(arc_preds[0], rel_preds[0], arcs, rels, mask, puncts, metric['1st'], batch) puncts = BiaffineSemanticDependencyParser.convert_to_3d_puncts( puncts, batch['mask_2nd']) # noinspection PyCallByClass BiaffineSemanticDependencyParser.update_metric( self, arc_preds[1], rel_preds[1], batch['arc_2nd'], batch['rel_2nd_id'], batch['mask_2nd'], puncts, metric['2nd'], batch)