https://github.com/Ruthvicp/CS5590_BigDataProgramming/blob/c3ab6a5f5750ba5c6646a931d977fc4d15cc8a3a/Lab/Lab4/Source/DecisionTree.py#L58-L59