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

Add OOB predictions to random forest #49

Open
mlondschien opened this issue Oct 3, 2021 · 0 comments
Open

Add OOB predictions to random forest #49

mlondschien opened this issue Oct 3, 2021 · 0 comments

Comments

@mlondschien
Copy link

I am using in-sample OOB predictions to estimate the KL-divergence between samples. In general, OOB predictions are an efficient alternative to CV to estimate out of sample prediction performance and can be used for tuning.

Getting OOB predictions requires storing the samples used to build each tree (i.e. indices here). This could be made optional. We can then add up predictions for samples only that were OOB for a particualr tree here, keeping track of the number of trees for which a particular sample was OOB.

I could work on a PR, but might need some help with details and guidance on what you think the API should be.

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

No branches or pull requests

1 participant