Skip to content

CausalML is not compatible with Scikit-Learn Pipeline #854

@PelinV

Description

@PelinV

Describe the solution you'd like
Swap the arguments treatment and y in the fit function.
To change the order of arguments in the fit function from X, treatment, y to X, y, treatment for compatibility with sklearn

Additional context
When using causalml models in the sklearn pipeline, it is necessary to write adapters
Because pipeline expects the following order: X, y, other arguments, and in fit in causalml the following order: X, treatment, y

Metadata

Metadata

Assignees

No one assigned

    Labels

    enhancementNew feature or request

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions