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.
"""
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)
