espnet.nets.e2e_asr_common.ErrorCalculator

source

class espnet.nets.e2e_asr_common.ErrorCalculator(char_list, sym_space, sym_blank, report_cer=False, report_wer=False)

Bases: object

Calculate CER and WER for E2E_ASR and CTC models during training.

• Parameters:
  • y_hats – numpy array with predicted text
  • y_pads – numpy array with true (target) text
  • char_list
  • sym_space
  • sym_blank
• Returns:

Construct an ErrorCalculator object.

calculate_cer(seqs_hat, seqs_true)

Calculate sentence-level CER score.

• Parameters:
  • seqs_hat (list) – prediction
  • seqs_true (list) – reference
• Returns: average sentence-level CER score

:rtype float

calculate_cer_ctc(ys_hat, ys_pad)

Calculate sentence-level CER score for CTC.

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

:rtype float

calculate_wer(seqs_hat, seqs_true)

Calculate sentence-level WER score.

• Parameters:
  • seqs_hat (list) – prediction
  • seqs_true (list) – reference
• Returns: average sentence-level WER score

:rtype float

convert_to_char(ys_hat, ys_pad)

Convert index to character.

• Parameters:
  • seqs_hat (torch.Tensor) – prediction (batch, seqlen)
  • seqs_true (torch.Tensor) – reference (batch, seqlen)
• Returns: token list of prediction

:rtype list :return: token list of reference :rtype list