Блог

Главная - Вопросы по программированию - Есть ли способ заставить export_graphviz заполнять узлы деревьев классификации, как если бы это было дерево регрессии?

Есть ли способ заставить export_graphviz заполнять узлы деревьев классификации, как если бы это было дерево регрессии?

#python #scikit-learn #graphviz #decision-tree

#python #scikit-learn #graphviz #дерево решений

Вопрос:

Я работаю с двоичной классификацией на очень несбалансированном наборе данных, и когда я экспортирую свое дерево решений в graphviz, используя export_graphviz и установив filled значение True все узлы дерева заполняются одним и тем же цветом. Есть ли способ изменить цвет с более высокой вероятностью принадлежности к альтернативному классу?

Ответ №1:

Вариант 1: Изменить вывод точек

Была бы возможность (возможно, автоматически) изменить вывод dot? Как только вы сможете идентифицировать рассматриваемые узлы, вы могли бы добавить (или адаптировать) атрибут fillcolor для этих узлов.

Вариант 2: настройка в sklearn

Я не знаю подробностей о sklearn. Но со страницы, на которую вы ссылаетесь, многие настройки кажутся интересными, особенно эти:

  • примесь
  • метка
  • class_names