adelie icon indicating copy to clipboard operation
adelie copied to clipboard

scikit compatible learners

Open apoorvalal opened this issue 7 months ago • 6 comments

Hi James,

great work on this. Was wondering if you were planning to provide a scikit-learn style API for this package? Would largely involve writing init/fit/predict/score methods, which would then allow adelie to be used in a number of useful downstream tasks (e.g. what i'm interested in - nuisance function estimation for AIPW / Rlearner for ATE/CATE estimation in econml / doubleml). Celer is a closely related package that provides a scikit-style API.

Totally understand if this isn't a priority; I might take a crack at it and send you a PR in case you're open to contributions.

apoorvalal avatar Jul 15 '24 05:07 apoorvalal