как оценить модель раса нлу - PullRequest
0 голосов
/ 19 апреля 2019

Ошибка при оценке модели rasa nlu с использованием модуля run_evaluation в блокноте kaggle:

TypeError: classification_report() got an unexpected keyword argument 'output_dict'



from rasa_nlu.evaluate import run_evaluation

run_evaluation("nlu.md", model_directory)
expected output:


INFO:rasa_nlu.evaluate:There were some nlu intent classification errors. Use `--verbose` to show them in the log.
INFO:rasa_nlu.evaluate:Model prediction errors saved to errors.json.
INFO:rasa_nlu.evaluate:Confusion matrix, without normalization: 
[[10  1  0  0  0  0  0]
 [ 0 13  0  0  0  0  0]
 [ 0  0 10  0  0  0  0]
 [ 0  0  0  5  0  0  0]
 [ 0  0  0  0  6  0  0]
 [ 0  0  0  0  0 15  0]
 [ 0  0  0  0  0  0 25]]

1 Ответ

0 голосов
/ 19 июня 2019

Похоже, это связано с используемой вами версией scikit-learn (https://github.com/RasaHQ/rasa/issues/1742).. Попробуйте установить scikit-learn 0.20.2 или, что еще лучше, обновить до Rasa 1.x, выполнив pip install rasa.

...