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

Question

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

1

Я пытаюсь получить отчет о классификации в виде словаря. Так что согласно документации 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.

Bidisha Das 10 окт. 2018, в 14:21

Источник

Теги:

python

python-3.x

scikit-learn

svm

1 ответ

Ещё вопросы

spundian · Answer 1 · 2018-10-19T16-37-00.000Z

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

import sklearn
print(sklearn.__version__)

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