Python3 implementation of ID3 algorithm based on categorical datasets.
- The tree is created using
DecisionTree.pyfile. - Its implementation of ID3 algorithm and returns the resulting tree as a multi-dimensional dictionary.
- Works on categorical datasets only.
- Unpredictable outputs are denoted by ?
- Added
DrawTree.pyto generate decision tree png image usingpyDotlibrary
- Open the terminal or command prompt. Head for the directory with source code
- PyDot package is based on GraphViz. Get the instructions to install it here
- Install python requirements:
pip install -r requirements.txt
Output of Test Cases:
Weather.csv
| Entry | Output |
|---|---|
| entry1 | no |
| entry2 | no |
| entry3 | yes |
| entry4 | yes |
| entry5 | no |
Vote.csv
| Entry | Output |
|---|---|
| entry1 | democrat |
| entry2 | democrat |
| entry3 | republican |
| entry4 | democrat |
| entry5 | democrat |
Soybean.csv
| Entry | Output |
|---|---|
| entry1 | ? |
| entry2 | frog-eye-leaf-spot |
| entry3 | frog-eye-leaf-spot |
| entry4 | brown-spot |
| entry5 | ? |
Feel free to leave any comments (as issues) or make suggestions (as a pull request).
React
Typescript
Django
Laravel
Facebook
Microsoft
Google
Alibaba
D3
Tencent