Skip to content

Support for random forest propensity score model? #91

Description

@jarbet

Random forests (RF) are a very simple and flexible nonparametric method for estimating the propensity score. They automatically allow for interactions, non-linear effects, can handle high-dimensional features, etc. RF tends to work well with default values too.

Currently weightit supports bart and gbm which also have the above benefits. I'm finding that the gbm method is too slow for large datasets. bart still works reasonably fast for large datasets but is more of a black box to me compared to RF. In any case, might be worth adding RF as an alternative approach?

EDIT: In the meantime, here is a slow workaround to use randomForest:

weightit(
    ...
    method = 'super',
    SL.library = c("SL.randomForest")
)

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions