Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Remove need for nrAssignments in DecisionTreeFactor::prune #1555

Merged
merged 4 commits into from
Jul 10, 2023

Conversation

varunagrawal
Copy link
Collaborator

@varunagrawal varunagrawal commented Jun 28, 2023

Use the DecisionTreeFactor::enumerate method to get all the probability values under consideration for pruning.

This eschews the need for nrAssignments, which means we could can get rid of it since it is not used anywhere else. I introduced this variable for this use case, but can't get it to work correctly since the value when computing DecisionTree::apply isn't well defined.

Hence this PR also removes nrAssignments.

@varunagrawal varunagrawal self-assigned this Jun 28, 2023
@varunagrawal varunagrawal merged commit e5a7bac into fix-1496 Jul 10, 2023
26 checks passed
@varunagrawal varunagrawal deleted the remove-nrAssignments branch July 10, 2023 23:40
@varunagrawal varunagrawal restored the remove-nrAssignments branch November 6, 2023 22:29
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.

1 participant