Skip to content

Conversation

rflperry
Copy link
Collaborator

@rflperry rflperry commented Mar 9, 2021

Reference issue

None

Type of change

Feature request

What does this implement/fix?

Adds additional parameters to Uncertainty Forest to better match the generalized random forest implementations in econml, the R grf, and the theory in general. Also parallelizes the learning of the decision trees in order to speed up fitting. The added parameters are as follows:

  • max_features (the number of features considered at each split node. Was all features before but most random forests use less than that by default)
  • poisson_sampler (technical detail to match the grf theory)
  • n_jobs (parallelization)
  • max_samples : (the number of samples subsampled for use in each tree. Was all samples before but conventional honest forests use 0.5*n_samples by default)

@rflperry rflperry added the draft In progress PR label Mar 9, 2021
@rflperry rflperry changed the title working prototype UF enhanced parameters to match GRF Mar 9, 2021
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.

black formatting is needed on the files changed~

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

Labels

draft In progress PR

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants