Skip to content

Conversation

@OliverSchacht
Copy link
Member

@OliverSchacht OliverSchacht commented Dec 2, 2024

Description

With this PR we are adding estimators for treatment effects in regression discontinuity designs. The estimators enable covariate adjustments powered by ML. We add a module doubleml.rdd that contains the estimator class RDFlex that follows usual package syntax. We also add synthetic DGPs in rdd.datasets. Finally, we include doubleml.rdd.global_learners which adds the functionality to ignore sample weights in scikit-learn estimators. This is useful to combine global and local estimators in RDD.
The new features are explained furtherly in the added documentation notebook.

Reference

The implementation follows this paper.

PR Checklist

  • The title of the pull request summarizes the changes made.
  • The PR contains a detailed description of all changes and additions.
  • References to related issues or PRs are added.
  • The code passes all (unit) tests.
  • Enhancements or new feature are equipped with unit tests.
  • The changes adhere to the PEP8 standards.

Copy link
Member

@SvenKlaassen SvenKlaassen left a comment

Choose a reason for hiding this comment

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

Just update the docstring of the simple dgp

Copy link
Member

Choose a reason for hiding this comment

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

Do we want to remove the area yield dgp from the import until we have a tuned version (and example in the gallery)?
Other option is to leave it in here and open a new branch for tuning and updating the docstring

Copy link
Member Author

Choose a reason for hiding this comment

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

Maybe we want to exclude this DGP in a release until we have a notebook describing it?

Copy link
Member

Choose a reason for hiding this comment

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

yes I think this would be the best option

@OliverSchacht OliverSchacht merged commit 60f4a43 into main Dec 3, 2024
11 checks passed
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.

4 participants