Task: Extract Decision Tree Split Attributes and Information Gain from Scikit-learn
Task: Extract Decision Tree Split Attributes and Information Gain from Scikit-learn: a task in Terminal-Lego-15k (Harbor dataset). Write a Python script that trains a decision tree classifier and extracts detailed information about which attributes (features) the tree splits on at each node, along…
The task
Write a Python script that trains a decision tree classifier and extracts detailed information about which attributes (features) the tree splits on at each node, along with their associated impurity/gain values. The script should demonstrate this for both `gini` and `entropy` splitting criteria, showing where the two…
Part of PrimeIntellect/Terminal-Lego-15k.