运行scikit-learn官方V-measure示例报错：v_measure_score不识别beta参数

错误原因 #

这个报错不是scikit-learn的Bug，而是本地scikit-learn版本与官方文档版本不兼容导致的：

• 官方稳定版文档中的v_measure_score已支持beta参数，但该参数是在scikit-learn 0.22版本之后才被添加到v_measure_score函数中的。
• 若你的本地版本低于0.22，v_measure_score确实没有beta这个入参，因此会触发该错误。

解决步骤 #

1. 检查当前scikit-learn版本
   运行以下代码确认版本：

   import sklearn
   print(sklearn.__version__)
   

2. 根据版本选择解决方案

   • 方案1：升级scikit-learn（推荐）
     执行命令升级到最新稳定版，即可直接使用文档中的代码：

     pip install --upgrade scikit-learn
     

   • 方案2：适配旧版本（不升级的情况）
     在0.22之前的版本中，调整beta参数需要使用metrics.v_measure函数（该函数返回同质性、完整性、V-measure三个值，取第三个即为目标结果）：

     from sklearn import metrics
     labels_true = [0, 0, 0, 1, 1, 1]
     labels_pred = [0, 0, 1, 1, 2, 2]
     _, _, v_score = metrics.v_measure(labels_true, labels_pred, beta=0.6)
     print(v_score)
     

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