Decision Tree Visualizer
v1.0.1
Визуализатор дерева решений - это мощная библиотека, которая позволяет вам легко визуализировать классификаторы дерева решений Sklearn. Он предоставляет функции для извлечения полезной информации о структуре и правилах дерева, и генерирует HTML -файлы для визуализации дерева решений.
Чтобы установить библиотеку, используйте PIP:
pip install d-treevis
Для начала импортируйте библиотеку и используйте функции create_tree и create_sankey :
import d_treevis as dtv Далее, вписывайте классификатор Tree Tree Tree Sklearn в свой набор данных и передайте его функции create_tree :
from sklearn . tree import DecisionTreeClassifier
# Fit a Decision Tree Classifier on your dataset
tree_model = DecisionTreeClassifier ()
tree_model . fit ( X , y )
# Visualize the decision tree
visualizer . visualize ( tree_model )Вы также можете настроить целевые имена и цвета для лучшей визуализации:
# Define target names and colors
target_names = [ 'Survived' , 'Not Survived' ]
target_colors = [ 'red' , 'yellow' ]
# Visualize the decision tree with custom target names and colors
visualizer . visualize ( tree_model , target_names = target_names , target_colors = target_colors )Для получения подробной документации и примеров, пожалуйста, обратитесь к документации «Визуализатор дерева решений».
Эта библиотека лицензирована по лицензии MIT.
Взносы приветствуются! Пожалуйста, прочитайте рекомендации по взносу перед отправкой запроса на привлечение.
Если у вас есть какие -либо вопросы или предложения, не стесняйтесь обращаться к нам по адресу [email protected].
