Skip to content

n_jobs not passed to 'auto' learners in SparseLinearDML() #1009

@hetankevin

Description

@hetankevin

When fitting a SparseLinearDML model, one can pass an n_jobs argument to the constructor to specify the number of cores used when fitting. However, this only applies to the LASSO second stage. When using the default 'auto' first stage model selector, the n_jobs argument is not passed to the first stage. This yields the unfortunate consequence of fitting random forests for both the treatment propensity and outcome models with only one CPU core enabled if the default auto first stage model selector is used.

This presumably is also an issue for the DML parent class.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions