编辑距离 cer wer

import editdistance  
  
def calculate_wer_cer(ref_text, pred_text,show=False):  
    # 计算WER  
    wer = editdistance.eval(ref_text, pred_text) / len(ref_text)  
    if(show):
        print(ref_text, pred_text,wer,'wer')  

    return wer
  
    # 计算CER  
    # cer = editdistance.eval(ref_text, pred_text) / len(ref_text.replace(" ", ""))  
    # print(f"CER: {cer:.2f}")  
  
# 参考文本和预测文本  
ref_text = "这是一个测试文本。包含一些测试数据。"  
pred_text = "这是一个测试文。断包含一一些测试数据。"  
  
# 计算WER和CER  
calculate_wer_cer(ref_text, pred_text)

计算字符错误率