Decision trees

Check decision tree creation here: https://planetcalc.com/8443/

Requirements: pandas and numpy

Run the tree:

python main.py

Run tests:

python -m pytest tests

In the future it would be better to create a framework that allows to create a specific type for the decision tree query.

Diagrams

From example.csv:

graph TB
    node_Outlook_("Outlook")
    node_Humidity_Sunny("Humidity")
    leaf_No_High["Play?: No"]
    leaf_Yes_Normal["Play?: Yes"]
    leaf_Yes_Overcast["Play?: Yes"]
    node_Windy_Rainy("Windy")
    leaf_Yes_False["Play?: Yes"]
    leaf_No_True["Play?: No"]

    node_Outlook_ -->|Sunny| node_Humidity_Sunny
    node_Humidity_Sunny -->|High| leaf_No_High
    node_Humidity_Sunny -->|Normal| leaf_Yes_Normal
    node_Outlook_ -->|Overcast| leaf_Yes_Overcast
    node_Outlook_ -->|Rainy| node_Windy_Rainy
    node_Windy_Rainy -->|False| leaf_Yes_False
    node_Windy_Rainy -->|True| leaf_No_True

Loading

From id3.csv

graph TB
    node_short_of_breath_("short_of_breath")
    node_fever_No("fever")
    leaf_No_No["Play?: No"]
    node_cough_Yes("cough")
    leaf_prob_No_2_Yes_1_Yes(["Play?: No: 2,Yes: 1"])
    node_fever_Yes("fever")
    node_cough_No("cough")
    leaf_prob_Yes_2_No_1_Yes(["Play?: Yes: 2,No: 1"])
    leaf_Yes_Yes["Play?: Yes"]

    node_short_of_breath_ -->|No| node_fever_No
    node_fever_No -->|No| leaf_No_No
    node_fever_No -->|Yes| node_cough_Yes
    node_cough_Yes -->|Yes| leaf_prob_No_2_Yes_1_Yes
    node_short_of_breath_ -->|Yes| node_fever_Yes
    node_fever_Yes -->|No| node_cough_No
    node_cough_No -->|Yes| leaf_prob_Yes_2_No_1_Yes
    node_fever_Yes -->|Yes| leaf_Yes_Yes

Loading