espnet2.legacy.nets.e2e_mt_common.ErrorCalculator

source

class espnet2.legacy.nets.e2e_mt_common.ErrorCalculator(char_list, sym_space, sym_pad, report_bleu=False)

Bases: object

Calculate BLEU for ST and MT models during training.

• Parameters:
  • y_hats – numpy array with predicted text
  • y_pads – numpy array with true (target) text
  • char_list – vocabulary list
  • sym_space – space symbol
  • sym_pad – pad symbol
  • report_bleu – report BLUE score if True

Construct an ErrorCalculator object.

calculate_bleu_ctc(ys_hat, ys_pad)

Calculate sentence-level BLEU score for CTC.

• Parameters:
  • ys_hat (torch.Tensor) – prediction (batch, seqlen)
  • ys_pad (torch.Tensor) – reference (batch, seqlen)
• Returns: corpus-level BLEU score

:rtype float

calculate_corpus_bleu(ys_hat, ys_pad)

Calculate corpus-level BLEU score in a mini-batch.

• Parameters:
  • seqs_hat (torch.Tensor) – prediction (batch, seqlen)
  • seqs_true (torch.Tensor) – reference (batch, seqlen)
• Returns: corpus-level BLEU score

:rtype float