使用run_evaluation评估Rasa NLU模型时遇TypeError及预期输出说明

嘿，这个问题我之前也碰到过，根源很明确——是scikit-learn版本不兼容导致的！

你看到的TypeError: classification_report() got an unexpected keyword argument 'output_dict'错误，是因为你当前使用的scikit-learn版本太旧了。output_dict这个参数是在scikit-learn 0.20.0版本才新增的，而Rasa NLU的run_evaluation模块在生成评估报告时需要用到这个参数，旧版本的scikit-learn不支持它，所以就报错了。

给你两个具体的解决步骤：

• 先升级scikit-learn到符合要求的版本，执行下面的命令就行：

  pip install --upgrade scikit-learn>=0.20.0
  

• 如果升级scikit-learn后还是有问题，那可能是Rasa NLU版本和新的scikit-learn不兼容，这时候可以尝试升级Rasa NLU到稳定版本：

  pip install --upgrade rasa-nlu
  

升级完成后，重新运行你的评估代码：

from rasa_nlu.evaluate import run_evaluation
run_evaluation("nlu.md", model_directory)

应该就能正常输出你预期的日志信息和混淆矩阵了。

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