def handle(self, t_input: sockeye.inference.TranslatorInput, t_output: sockeye.inference.TranslatorOutput): """ :param t_input: Translator input. :param t_output: Translator output. """ plot_attention(t_output.attention_matrix, t_input.tokens, t_output.tokens, "%s_%d.png" % (self.plot_prefix, t_input.id))
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. """ plot_attention(t_output.attention_matrix, t_input.tokens, t_output.tokens, "%s_%d.png" % (self.plot_prefix, t_input.sentence_id))
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: plot_attention(t_output.attention_matrix, t_input.tokens, t_output.tokens, "%s_%d.png" % (self.plot_prefix, t_input.id)) elif self.output_attention_type == C.ENCODER_SELF_ATTENTION: plot_attention(t_output.attention_matrix, t_input.tokens, t_input.tokens, "%s_%d.png" % (self.plot_prefix, t_input.id)) else: plot_attention(t_output.attention_matrix, ['<s>'] + t_output.tokens, t_output.tokens, "%s_%d.png" % (self.plot_prefix, t_input.id))