TheDigitalFrontier / parallel-decision-trees

Semester project in CS205 Computing Foundations for Computational Science at Harvard School of Engineering and Applied Sciences, spring 2020.
MIT License
3 stars 1 forks source link

Reorganize code for stopping conditions. #56

Closed gpestre closed 4 years ago

gpestre commented 4 years ago

Mostly updated for clarity (using "return" keyword when a pruning condition is met, rather than just skipping over the splitting code).

But also fixed a minor bug: The previous code incremented the leaf counter too soon (i.e. there was a possibility it would get incremented even if we ended up pruning because one side of the split had zero observations).