https://github.com/yashksaini-coder/decision-tree-visualizer
https://github.com/yashksaini-coder/decision-tree-visualizer
decision-trees pypi pypi-package visualization
Last synced: 4 months ago
JSON representation
- Host: GitHub
- URL: https://github.com/yashksaini-coder/decision-tree-visualizer
- Owner: yashksaini-coder
- License: apache-2.0
- Created: 2024-10-24T14:12:23.000Z (8 months ago)
- Default Branch: master
- Last Pushed: 2024-10-24T15:26:06.000Z (8 months ago)
- Last Synced: 2024-10-25T17:49:30.588Z (8 months ago)
- Topics: decision-trees, pypi, pypi-package, visualization
- Language: Python
- Homepage: https://pypi.org/project/d-treevis/
- Size: 263 KB
- Stars: 1
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# 📚🔍🎨 **Decision Tree Visualizer**
## **Introduction**
The Decision Tree Visualizer is a powerful library that allows you to visualize sklearn Decision Tree Classifiers with ease. It provides functions for extracting useful information about the tree structure and rules, and generates HTML files for visualizing the decision tree.
## 📊 **Features**
- 🌳 Visualize sklearn Decision Tree Classifiers using HTML templates
- 🔍 Extract useful information about the tree structure and rules
- 📊 Generate output HTML files for visualization
- 🎨 Customize target names and colors for better visualization
## 🔧 **Installation**
To install the library, use pip:
```
pip install d-treevis
```
## 📖 **Usage**
To get started, import the library and use the `create_tree` and `create_sankey` functions:
```python
import d_treevis as dtv
```
Next, fit a sklearn Decision Tree Classifier on your dataset and pass it to the `create_tree` function:
```python
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)
```
You can also customize the target names and colors for better visualization:
```python
# 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)
```
## 📚 **Documentation**
For detailed documentation and examples, please refer to the [Decision Tree Visualizer Documentation](https://example.com/documentation).
## 📄 **License**
This library is licensed under the [MIT License](https://example.com/license).
## 🙌 **Contributing**
Contributions are welcome! Please read the [Contribution Guidelines](https://example.com/contributing) before submitting a pull request.
## 📧 **Contact**
If you have any questions or suggestions, feel free to reach out to us at [[email protected]](mailto:[email protected]).
## 🔖 **References**
- [sklearn Decision Tree Classifier](https://scikit-learn.org/stable/modules/generated/sklearn.tree.DecisionTreeClassifier.html)