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

Refactor to avoid inheritance in BaseNode #35

Merged

Conversation

fjloyola
Copy link
Contributor

@fjloyola fjloyola commented Jan 7, 2023

Refactor of the classes BaseNode, SplitNode and LeafNode to avoid using inheritance into just one class Node, so we could use JitClasses of numba in the near future;

…ng inheritance into just one class Node, so we could use JitClasses of numba in the near future;

Change the behaviour of _traverse_tree to return the leaf node value or the mean of leaf nodes values;
@fjloyola fjloyola force-pushed the feature/tree-class-without-inheritance branch from 762c505 to d7ef4cb Compare January 8, 2023 15:57
@fjloyola fjloyola marked this pull request as ready for review January 9, 2023 01:54
@aloctavodia aloctavodia merged commit 6aad349 into pymc-devs:main Jan 9, 2023
@fjloyola fjloyola deleted the feature/tree-class-without-inheritance branch January 9, 2023 12:27
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.

None yet

2 participants