Как получить вывод sklearn.metrics.classification_report в качестве диктата?

Question

Я пытался получить отчет о классификации в виде словаря.Таким образом, в соответствии с документацией scikit-learn 0.20, я делаю:

from sklearn import metrics

rep = metrics.classification_report(y_true, y_pred, output_dict=True)

Но получаю сообщение об ошибке

TypeError: classification_report() got an unexpected keyword argument 'output_dict'
The scikit-learn module in my machine was initially 0.19.1 but even after updating it to 0.20, the same error message shows.

Answer 1

Эта ошибка не должна появляться, если у вас установлен scikit-learn 0.20.0.Если вы пытаетесь сделать это в блокноте jupyter, убедитесь, что в вашем блокноте отображается правильная версия, используя:

import sklearn
print(sklearn.__version__)

Если вы обновили scikit-learn, но jupyter показывает неверную версию пакета, убедитесь, чтоjupyter установлен в вашей текущей среде (и перезапустите jupyter в новом терминале).