Attribute selection in Decision tree Algorithm

Question

I'm trying to figure out how the decision tree algorithm works. I had a small doubt. How do I select attributes for the tree?

Answer 1

If you have a dataset that consists of n attributes, then selecting what should be placed at the root or different levels of the three can get confusing. You can use matrices like information gain, index, etc. These will calculate values for every attribute and these values are stored and the attribute order is decided. 

While using information Gain as a criterion, we assume attributes to be categorical, and for Gini index, attributes are assumed to be continuous.

Discover the Power of AI, Enroll now in the AI in Finance course today!

Comments

What is the information gain?

---

Refer to my answer here

https://www.edureka.co/community/46267/what-is-information-gain-decision-tree-algo