Skip to content

D0CRT Accept Scikit-learn models directly instead of parameter dictionaries #369

@jpaillard

Description

@jpaillard

Working on #253, I realize that a lot of code would need to be rewritten to adapt to the classification case, while not much is changing in theory.
One of the reasons for that is the use of dictionaries of parameters (like params_lasso_distillation_x) and the method: _fit_lasso, which are incompatible with LogisticRegression.

I suggest replacing these dictionaries with Scikit-learn models.

Metadata

Metadata

Assignees

No one assigned

    Labels

    estimatorsquestion link to estimatorsmethod implementationQuestion regarding methods implementations

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions