def Y_to_outputs(self, Y: Union[tf.Tensor, Tuple[tf.Tensor]], gold=False, inputs=None, X=None) -> Iterable: for words, tags in zip( inputs, super().Y_to_outputs(Y, gold, inputs=inputs, X=X)): yield from iobes_to_span(words, tags)
def predict_batch(self, batch, inputs=None): for words, tags in zip(inputs, super().predict_batch(batch, inputs)): yield from iobes_to_span(words, tags)