实现Hugging Face MT5摘要代码遇AttributeError问题求助

问题原因 #

错误源于rouge评分工具的版本更新：旧版本返回带mid属性的评分对象，新版本直接返回numpy.float64类型的数值，导致原代码调用value.mid.fmeasure时找不到对应属性。

修复方案 #

修改计算结果的代码，直接使用返回的数值即可：

原代码：

result = {key: value.mid.fmeasure * 100 for key, value in result.items()}

修改为：

result = {key: value * 100 for key, value in result.items()}

额外验证 #

若想确认评分结构，可先打印result查看：

print(result)

新版rouge返回结果类似{'rouge1': 0.456, 'rouge2': 0.321, 'rougeL': 0.412}，直接取这些数值计算即可。

内容的提问来源于stack exchange，提问作者phoebephalange