Skip to content

Fix Information Gain Calculation in Decision Tree Algorithm#2

Open
ElmekaouiHaitham wants to merge 1 commit intoParam302:masterfrom
ElmekaouiHaitham:patch-1
Open

Fix Information Gain Calculation in Decision Tree Algorithm#2
ElmekaouiHaitham wants to merge 1 commit intoParam302:masterfrom
ElmekaouiHaitham:patch-1

Conversation

@ElmekaouiHaitham
Copy link

This pull request corrects the formula for calculating Information Gain in the decision tree algorithm. The previous formula was incorrect as it incorrectly applied the root entropy to the weighted purities of the left and right child nodes. The corrected formula now properly calculates Information Gain by multiplying the root entropy with the weighted sum of the entropies of the left and right child nodes.

in the information gain formula it is the entropy, not the purity
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants