mirror of
https://github.com/open-compass/opencompass.git
synced 2025-05-30 16:03:24 +08:00
32 lines
838 B
Python
32 lines
838 B
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
import jieba
|
||
|
|
from nltk import bleu
|
||
|
|
from nltk.translate.bleu_score import SmoothingFunction
|
||
|
|
from rouge import Rouge
|
||
|
|
|
||
|
|
def is_chinese(text):
|
||
|
|
for char in text:
|
||
|
|
if '\u4e00' <= char <= '\u9fff':
|
||
|
|
return True
|
||
|
|
return False
|
||
|
|
|
||
|
|
def compute_acc(gt_list, pred_list):
|
||
|
|
rouge_l = 0
|
||
|
|
rouge = Rouge()
|
||
|
|
|
||
|
|
for pred, gold in zip(pred_list, gt_list):
|
||
|
|
if is_chinese(pred):
|
||
|
|
prediction = " ".join(jieba.cut(pred))
|
||
|
|
gold = " ".join(jieba.cut(gold))
|
||
|
|
else:
|
||
|
|
prediction = pred
|
||
|
|
gold = gold
|
||
|
|
|
||
|
|
try:
|
||
|
|
scores = rouge.get_scores(prediction, gold)
|
||
|
|
rouge_l += scores[0]['rouge-l']['r']
|
||
|
|
except:
|
||
|
|
continue
|
||
|
|
avg_rougel = rouge_l / len(gt_list)
|
||
|
|
return avg_rougel
|