[Community Puzzle] Basic Decision Tree - 2


#1

https://www.codingame.com/training/hard/basic-decision-tree---2

Send your feedback or ask for help here!

Created by @Wei-1,validated by @bbb000bbbyyy.
If you have any issues, feel free to ping them.


#2

I have got an issue on test 5:
I have got it right for features 1 2 and 3, but my method goes further and add feature 4 at the end.
And manually I get the same result:
after filtering with variable 1 2 and 3 the remaining un classified data are:
index output f4
0 1 1
6 3 2
7 3 1
8 3 1

adding a node with f4, decrease entropy (I have 0.81 on the list of labels [1,3,3,3] and 0.69 on the 2 sets [3] and [1,3,3]).

Can you tell me where I might be wrong ?

Thank you for your time,

Mathieu