class_names в export_graphviz непредвиденная ошибка ключевого слова

Со следующим кодом:

classes = ['ham', 'spam']
dot_data = StringIO()
with open("./plots/ritesh.dot", "w") as f:
    export_graphviz(dt_old.named_steps['classifier'], out_file=f, feature_names=vocab1, class_names=classes)

Я получаю сообщение об ошибке как неожиданный аргумент ключевого слова.

Traceback (most recent call last):
  File "DecisionTree.py", line 198, in <module>
    main()
  File "DecisionTree.py", line 133, in main
    export_graphviz(dt_old.named_steps['classifier'], out_file=f, feature_names=vocab1, class_names=classes)
TypeError: export_graphviz() got an unexpected keyword argument 'class_names'

Но в документе scikit-learn четко указано, что

sklearn.tree.export_graphviz(decision_tree, out_file='tree.dot', max_depth=None, feature_names=None, class_names=None, label='all', filled=False, leaves_parallel=False, impurity=True, node_ids=False, proportion=False, rotate=False, rounded=False, special_characters=False)

В чем причина этого TypeError?


python-2.7 scikit-learn graphviz pydot
person Zangetsu    schedule 01.12.2015    source источник

Ответы (1)


arrow_upward
7
arrow_downward

Установленная вами версия sklearn, вероятно, устарела с документацией. Если вы установили sklearn с помощью anaconda, попробуйте ввести в командной строке «conda update scikit-learn».

person nadnerbgalea    schedule 07.12.2015