Как найти индекс каждого листа или узла в дереве решений? - PullRequest
0 голосов
/ 12 апреля 2020

Основной вопрос состоит в том, чтобы найти, к какому листовому узлу относится каждая выборка. Есть тысячи сообщений об использовании tree.apply. Мне хорошо известна эта функция, которая возвращает индекс листового узла.

Теперь я хотел бы добавить листовой индекс в узлах графа (который я генерирую с помощью Graphviz).

Рисование метода перечисления, используемого для индексов, не будет работать. Дерево решений, которое я разрабатываю, довольно большое. Поэтому мне нужно иметь возможность печатать листовой индекс на графике.

Другой вариант, к которому я открыт, - это создать массив со всеми листовыми индексами (в том же порядке) листовых узлов дерево решений. Любой намек на то, как это сделать?

1 Ответ

0 голосов
/ 12 апреля 2020

Имеется параметр node_ids команды export_graphviz. Если для этого параметра задано значение True, то индексы добавляются в метку дерева решений.

...