Learn Before
Concept
Information Gain
We define information gain as .
We interpret information gain as the amount of information we learn about the label , given a specific feature .
Note that if the feature is completely uncorrelated with the label , then . Therefore .
As such when we choose the next feature to split on we should choose the one that maximizes information gained.
0
1
Updated 2020-03-15
Tags
Data Science