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

[ENH] Adding leaf node samples to be stored when "quantile" tree is turned on #45

Merged
merged 21 commits into from
Jun 15, 2023

Conversation

adam2392
Copy link
Collaborator

Reference Issues/PRs

Addresses the quantile-trees part of: neurodata/treeple#29

What does this implement/fix? Explain your changes.

  1. Stores for each leaf node a 2D numpy array of the y-samples (remember y is (n_samples, n_outputs))
  2. Does this all the way in Criterion
  3. Only supports supervised tree/splitter/criterion

Any other comments?

Signed-off-by: Adam Li <adam2392@gmail.com>
@adam2392
Copy link
Collaborator Author

@sampan501 and @PSSF23 you both might be interested.

This is a WIP, but almost done addition of code into the sklearn-fork to enable all trees to optionally store all leaf-node samples. This is useful for:

  1. quantile regression
  2. implementing procedures where we use samples in the leaves (e.g. trying to marginalize a forest, etc.)

@adam2392 adam2392 merged commit 21ccb30 into submodulev2 Jun 15, 2023
1 of 4 checks passed
@adam2392 adam2392 deleted the quantiles branch June 15, 2023 14:35
Copy link
Member

@PSSF23 PSSF23 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@adam2392 predict_quantiles functions have check_input documentation but the param is not in the function. Meanwhile get_leaf_node_samples has the param but it's not documented. These might cause the errors below:

https://app.circleci.com/pipelines/github/neurodata/scikit-tree/463/workflows/b92f9a94-dbb3-4b9d-b1f0-5a75c9941062/jobs/771

@adam2392
Copy link
Collaborator Author

I pushed a commit 3f5cb65

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

8 participants