def handle(self, t_input: sockeye.inference.TranslatorInput, t_output: sockeye.inference.TranslatorOutput): """ :param t_input: Translator input. :param t_output: Translator output. """ print_attention_text(t_output.attention_matrix, t_input.tokens, t_output.tokens, self.threshold)
def handle(self, t_input: inference.TranslatorInput, t_output: inference.TranslatorOutput, t_walltime: float = 0.): """ :param t_input: Translator input. :param t_output: Translator output. :param t_walltime: Total wall-clock time for translation. """ print_attention_text(t_output.attention_matrix, t_input.tokens, t_output.tokens, self.threshold)
def handle(self, t_input: inference.TranslatorInput, t_output: inference.TranslatorOutput, t_walltime: float = 0.): """ :param t_input: Translator input. :param t_output: Translator output. :param t_walltime: Total wall-clock time for translation. """ if self.output_attention_type == C.ENCODER_DECODER_ATTENTION: print_attention_text(t_output.attention_matrix, t_input.tokens, t_output.tokens, self.threshold) elif self.output_attention_type == C.ENCODER_SELF_ATTENTION: print_attention_text(t_output.attention_matrix, t_input.tokens, t_input.tokens, self.threshold) else: print_attention_text(t_output.attention_matrix, ['<s>'] + t_output.tokens, t_output.tokens, self.threshold)