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

Question

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

Answer 1

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

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

Вариант 2: конфигурация в sklearn

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

• примесь
• метка
• имя_класса