Machine Learning & AI
Intermediate

Information Gain

Reduction in entropy achieved by splitting a node on a feature.

Formula

IG=H(S)kSkSH(Sk)IG = H(S) - \sum_k \dfrac{|S_k|}{|S|}H(S_k)

Variables

H(S)Parent entropy
S_kChild subsets

Example

Split drops H from 1.0 to 0.5: IG = 0.5

Did You Know?

Decision trees greedily pick the split with the highest information gain at each step.

Share this formula